问如何查看测试准确度，用对抗自编码器时？

for x,y in test_loader:

x=x.view(x.size(0),3*64*64)

out=autoencoder(x)

pred=out.argmax(dim=1)

correct=pred.eq(y).sum().float().item()

total_correct+=correct

total_num=len(test_loader.dataset)

acc=total_correct/total_num

print('test acc:',acc)

我用以上代码告诉我没有属性‘argmax()’,请问博主这个怎么解决，如果我想看到测试准确率？