tf.linalg.lu

Computes the LU decomposition of one or more square matrices.