RuntimeError: view size is not compatible完美解决

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。