File "/home/yuanlina/PycharmProjects/ViT-pytorch-main/test1-10.py", line 136, in Accuracy
correct_k = correct[:k].view(-1).float().sum(0, keepdim=True)
correct_k = correct[:k].contiguous().view(-1).float().sum(0, keepdim=True)
correct_k = correct[:k].contiguous().view(-1).float().sum(0, keepdim=True)
添加.contiguous()即可