tensorflow lu ops related (test script)

lu_ops_test.py

import tensorflow as tf
import numpy as np
w = np.array([[3,-1,-1], [-1,5.0,-1], [1, 0 ,10]])
print(w)
W = tf.constant(w)

l, u, p, q = tf.lu(W)

pinv = tf.matrix_inverse(p)
qinv = tf.matrix_inverse(q)
a1 = tf.matmul(pinv, l)
a2 = tf.matmul(a1, u)
a = tf.matmul(a2, qinv)


with tf.Session() as sess:
    print('Lu')
    sess.run(l)
    print('l\n', l.eval())
    print('u\n', u.eval())
    print('q\n', q.eval())
    print('r\n', r.eval())
    sess.run(a)
    print('a\n', a.eval())