tf.keras.backend.l2_normalize

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

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

A tensor.