tf.compat.v1.linalg.l2_normalize