tf.contrib.layers.unit_norm( inputs, dim, epsilon=1e-07, scope=None )
Normalizes the given input across the specified dimension to unit length.
Note that the rank of
input must be known.
Tensorof 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.
ValueError: If dim is smaller than the number of dimensions in 'inputs'.
© 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.