tf.math.l2_normalize