问题遇到的现象和发生背景
问题相关代码,请勿粘贴截图
运行结果及报错内容
我的解答思路和尝试过的方法
我想要达到的结果
p = p.view(bs, self.na, self.no, self.ny, self.nx).permute(0, 1, 3, 4, 2).contiguous() 什么意思?
问题相关代码,请勿粘贴截图
p = p.view(bs, self.na, self.no, self.ny, self.nx).permute(0, 1, 3, 4, 2).contiguous() 什么意思?
p.view(bs, self.na, self.no, self.ny, self.nx)
生成一个多维度列表
permute(0, 1, 3, 4, 2)
将第二维度放置到最后位置,将3,4纬度前移一位
contiguous()
返回一个前面操作的强制拷贝对象,后面操作这个对象时不会对前面的数据发送改变
有帮助点个采纳,谢谢