acc_meter.update_state使用问题

Question

问题遇到的现象和发生背景

我看网上很多代码都有这样一段话：

for step,(x,y) in enumerate(test_data):
        out = model(x)
        pred = tf.argmax(out,axis=-1)
        pred = tf.cast(pred,dtype=tf.int32)
        y = tf.cast(tf.argmax(y,axis=-1),dtype=tf.int32)
        acc_meter.update_state(y,pred)

我的问题：可是pred返回的不是下标吗？怎么能作为一个预测值，不应该是y[pred]这种方式传进去吗？