tf.linalg.tensor_diag