Как я могу вычислить эту точность без зацикливания каждого элемента данных?

Question

Я использую цикл for для вычисления точности между y_pred и y_true, это делает мой код очень медленным, поэтому я хочу избежать этого для цикла for. Как я могу изменить этот код?

sumAcc =  tf.Variable(0, dtype=tf.float32)
for i in range(0, y_pred_batch_size):
    ele_y_pred = y_pred[i, 0:label_length[i,0]]
    ele_y_true = y_true[i, 0:label_length[i,0]]
    sumAcc =  sumAcc + tf.reduce_mean(tf.cast(tf.equal( ele_y_pred, ele_y_true), tf.float32)) 
tf.print("\n","Accuracy: ",sumAcc / tf.constant(y_pred_batch_size,dtype=tf.float32) * 100,"%" )

Значения y_pred и y_true выглядят как .. [batch_index, [[45],[56],[12],...]