tf.linalg.global_norm