RuntimeError: view size is not compatible with input tensor‘s size and stride

出现这个原因主要就是因为view()需要Tensor中的元素地址是连续的，因为可能出现Tensor不连续的情况，所以修改方法为：

在.view前加.contiguous()，使其变为连续就ok。

最后加上.contiguous()

for k in topk:
    # print(correct.shape)
    correct_k = correct[:k].contiguous().view(-1).float().sum(0, keepdim=True)
    wrong_k = batch_size - correct_k
    res.append(wrong_k.mul_(100.0 / batch_size))