tf.linalg.logm