如何解决Pytorch 尺寸变化
是否有任何方法可以将 [1,512,1,1] 更改为 [1,2,2] 张量。 我知道仅通过更改尺寸是不可能的。 有没有什么方法可以在 PyTorch 中使用 concat 或 stack（torch.stack、torch.cat）
我使用以下代码制作张量
a = torch.rand([1,1])
如何将其更改为维度为 [1,2] 的张量
解决方法
我试过了
tmp = torch.cat([a,a],2)
a = torch.cat([tmp,tmp],3)
那就是torch.repeat，这将复制数据：
>>> a = a.repeat(1,1,2,2)
如果您不想复制数据，请使用torch.expand：
>>> a = a.expand(-1,-1,2)
版权声明：本文内容由互联网用户自发贡献，该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容， 请发送邮件至 dio@foxmail.com 举报，一经查实，本站将立刻删除。
