pytorch Tensor 相减问题请教

2021-11-12 16:41:23 +08:00
 zxCoder
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]

571 次点击
所在节点    问与答
2 条回复
bravecarrot
2021-11-13 18:59:38 +08:00
broadcasting
ekidona
2021-11-14 13:35:11 +08:00
a.permute(1,0,2)-b

这是一个专为移动设备优化的页面(即为了让你能够在 Google 搜索结果里秒开这个页面),如果你希望参与 V2EX 社区的讨论,你可以继续到 V2EX 上打开本讨论主题的完整版本。

https://www.v2ex.com/t/814998

V2EX 是创意工作者们的社区,是一个分享自己正在做的有趣事物、交流想法,可以遇见新朋友甚至新机会的地方。

V2EX is a community of developers, designers and creative people.

© 2021 V2EX