tf.math.l2_normalize

Normalizes along dimension axis using an L2 norm.

For a 1-D tensor with axis = 0, computes

output = x / sqrt(max(sum(x**2), epsilon))

For x with more dimensions, independently normalizes each 1-D slice along dimension axis.

Args
x A Tensor.
axis Dimension along which to normalize. A scalar or a vector of integers.
epsilon A lower bound value for the norm. Will use sqrt(epsilon) as the divisor if norm < sqrt(epsilon).
name A name for this operation (optional).
Returns
A Tensor with the same shape as x.

© 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/r2.4/api_docs/python/tf/math/l2_normalize