问题: 比较2个tensor的时候,返回全部是FALSE,不符合预期 torch.argmax(y_hat, dim=1)==y 解决方法: y_idx=torch.argmax(y_hat, dim=1) y_idx==y