RuntimeError: view size is not compatible with input tensor's size and stride (at least one dimension spans across two contiguous subspaces). Use .reshape(…) instead.
报错在这一句：
correct_k = correct[:k].view(-1).float().sum(0)
加入.contiguous()即可，修改如下：
correct_k = correct[:k].contiguous().view(-1).float().sum(0)
这是因为view()需要Tensor中的元素地址是连续的，因为可能出现Tensor不连续的情况。
修改方法为：在.view前加
.contiguous()
，使其变为连续就ok。
最终效果如下：
参考链接：
一步真实解决RuntimeError: view size is not compatible with input tensor‘s size and stride-pudn.com
