Оптимизация кода внутри сеанса тензорного потока

Question

У меня есть класс, определенный как

class object:
    ....
    W = tf.Variable(tf.zeros([8000,1])) 
    Z = tf.Variable(tf.truncated_normal([8000,1], stddev=0.1))
    H = tf.zeros([8000,1])
    ....

Позже, когда я запускаю сеанс TensorFlow, приведенный ниже фрагмент кода занимает слишком много. Я не очень знаком с Tensorflow, поэтому мне было интересно, есть ли лучший способ заставить код работать быстрее

        y =  np.zeros([8000,1])
        for i in range(50):
            y[:4000] = (object[i].W[:4000] + rho*object[i].H[0]*object[i].Z[:4000]).eval()
            y[4000:] = (object[i].W[4000:] + rho*object[i].H[1]*object[i].Z[:4000]).eval()
            object[i].W.assign(y)

Thamk you!