tf.keras.backend.l2_normalize

TensorFlow 2 version View source on GitHub

Normalizes a tensor wrt the L2 norm alongside the specified axis.

x Tensor or variable.
axis axis along which to perform normalization.

A tensor.