tf.math.l2_normalize

View source on GitHub

Normalizes along dimension axis using an L2 norm.

tf.math.l2_normalize(
    x, axis=None, epsilon=1e-12, name=None
)

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:

Returns:

A Tensor with the same shape as x.