On this page
tf.contrib.layers.unit_norm
Normalizes the given input across the specified dimension to unit length.
tf.contrib.layers.unit_norm(
inputs, dim, epsilon=1e-07, scope=None
)
Note that the rank of input
must be known.
Args | |
---|---|
inputs |
A Tensor of arbitrary size. |
dim |
The dimension along which the input is normalized. |
epsilon |
A small value to add to the inputs to avoid dividing by zero. |
scope |
Optional scope for variable_scope. |
Returns | |
---|---|
The normalized Tensor . |
Raises | |
---|---|
ValueError |
If dim is smaller than the number of dimensions in 'inputs'. |
© 2020 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/versions/r1.15/api_docs/python/tf/contrib/layers/unit_norm