W3cubDocs

/TensorFlow Python

tf.contrib.layers.unit_norm(args, *kwargs)

tf.contrib.layers.unit_norm(*args, **kwargs)

See the guide: Layers (contrib) > Higher level ops for building neural network layers

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'.

Defined in tensorflow/contrib/framework/python/ops/arg_scope.py.

© 2017 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/contrib/layers/unit_norm