a=torch.Tensor([
[[1,1],[2,2],[3,3]],
[[4,4],[5,5],[6,6]],
])
b=torch.Tensor([
[1.5,2.5],
[3.5,4.5],
])
print(a.size()) # torch.Size([2, 3, 2])
print(b.size()) # torch.Size([2, 2])
c=a-b
有没有不需要 unsqueeze 和 repeat ,直接相减的方法,听说可以节省内存
就是想让[1,1],[2,2],[3,3]都减去[1.5,2.5],[4,4],[5,5],[6,6]都减去[3.5,4.5]
这是一个专为移动设备优化的页面(即为了让你能够在 Google 搜索结果里秒开这个页面),如果你希望参与 V2EX 社区的讨论,你可以继续到 V2EX 上打开本讨论主题的完整版本。
V2EX 是创意工作者们的社区,是一个分享自己正在做的有趣事物、交流想法,可以遇见新朋友甚至新机会的地方。
V2EX is a community of developers, designers and creative people.