tf.linalg.lu_matrix_inverse