RuntimeError: view size is not compatible with input tensor‘s size and stride解决办法

在跑SENET代码时：

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.

问题分析：

出现这个原因主要就是因为view()需要Tensor中的元素地址是连续的，因为可能出现Tensor不连续的情况，所以修改方法为：
在.view前加.contiguous（），使其变为连续就ok。

 

修改代码：

    for k in topk:
        correct_k = correct[:k].contiguous().view(-1).float().sum(0)
        res.append(correct_k.mul_(100.0 / batch_size))