pytorch Tensor 相减问题请教

V2EX = way to explore

V2EX 是一个关于分享和探索的地方

现在注册

已注册用户请登录

这是一个创建于 1132 天前的主题，其中的信息可能已经有所发展或是发生改变。

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]

Torch

tensor

size

2 条回复 • 2021-11-14 13:35:11 +08:00

bravecarrot

2021-11-13 18:59:38 +08:00

broadcasting

ekidona

2021-11-14 13:35:11 +08:00 via iPhone

a.permute(1,0,2)-b