tf.contrib.layers.unit_norm

Normalizes the given input across the specified dimension to unit length.

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