UnitNorm
Inherits From: Constraint
tf.keras.constraints.UnitNorm
tf.keras.constraints.unit_norm
Defined in tensorflow/python/keras/_impl/keras/constraints.py
.
Constrains the weights incident to each hidden unit to have unit norm.
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)
.__init__
__init__(axis=0)
Initialize self. See help(type(self)) for accurate signature.
__call__
__call__(w)
Call self as a function.
get_config
get_config()
© 2018 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/api_docs/python/tf/keras/constraints/UnitNorm