TensorFlow 1 version | View source on GitHub |
Constrains the weights incident to each hidden unit to have unit norm.
Inherits From: Constraint
tf.keras.constraints.UnitNorm(
axis=0
)
Also available via the shortcut function tf.keras.constraints.unit_norm
.
Arguments | |
---|---|
axis
|
integer, axis along which to calculate weight norms.
For instance, in a Dense layer the weight matrix
has shape (input_dim, output_dim) ,
set axis to 0 to constrain each weight vector
of length (input_dim,) .
In a Conv2D layer with data_format="channels_last" ,
the weight tensor has shape
(rows, cols, input_depth, output_depth) ,
set axis to [0, 1, 2]
to constrain the weights of each filter tensor of size
(rows, cols, input_depth) .
|