a constraint for a maximum norm of the weights
More...
a constraint for a maximum norm of the weights
§ __init__()
def nabu.neuralnetworks.components.constraints.MaxNorm.__init__ |
( |
|
self, |
|
|
|
max_norm = 1 , |
|
|
|
axis = 0 |
|
) |
| |
constructor
- Parameters
-
max_norm | the maximum norm |
axis | the axis to compute the norm |
§ __call__()
def nabu.neuralnetworks.components.constraints.MaxNorm.__call__ |
( |
|
self, |
|
|
|
tensor |
|
) |
| |
apply the constraint
- Parameters
-
tensor | the tensor to apply the constraint to |
The documentation for this class was generated from the following file: