🎨 Format Python code with psf/black

This commit is contained in:
ndem0
2024-02-09 11:25:00 +00:00
committed by Nicola Demo
parent 591aeeb02b
commit cbb43a5392
64 changed files with 1323 additions and 955 deletions

View File

@@ -3,7 +3,7 @@ from torch.nn.parameter import Parameter
class AdaptiveSoftplus(torch.nn.Module):
'''
"""
Implementation of soft exponential activation.
Shape:
- Input: (N, *) where * means, any number of additional
@@ -18,16 +18,16 @@ class AdaptiveSoftplus(torch.nn.Module):
>>> a1 = soft_exponential(256)
>>> x = torch.randn(256)
>>> x = a1(x)
'''
"""
def __init__(self):
'''
"""
Initialization.
INPUT:
- in_features: shape of the input
- aplha: trainable parameter
aplha is initialized with zero value by default
'''
"""
super().__init__()
self.soft = torch.nn.Softplus()
@@ -36,9 +36,9 @@ class AdaptiveSoftplus(torch.nn.Module):
self.scale.requiresGrad = True # set requiresGrad to true!
def forward(self, x):
'''
"""
Forward pass of the function.
Applies the function to the input elementwise.
'''
#x += self.translate
"""
# x += self.translate
return self.soft(x) * self.scale