错误原因:

我们使用view()函数的时候,稍有不慎会碰到这样的错误:

RuntimeError: invalid argument 2: view size is not compatible with input tensor's....

这是因为tensor经过转置后数据的内存地址不连续导致的。

解决方案:

在view()函数前加contiguous(),使其变为连续就可以了,如下所示:

out = out.contiguous().view(2, 3)

牢记:

目前pytorch只要是transpose()和permute()这两个函数用过后,tensor都会变得不再连续,就不可以使用view()。

我们只需要记住了,使用view()之前,只要使用了transpose()和permute()这两个函数一定要contiguous()。

调用contiguous()时,会强制拷贝一份tensor,让它的布局和从头创建的一模一样。

读者推荐

为了让大家更好的掌握PyTorch的基础和核心,推荐大家关注《PyTorch面试精华》