У меня есть два тензора Pytorch, a и b, формы (S, M) и (S, M, H) соответственно. M - размер моей партии. Я хочу умножить и суммировать два тензора, чтобы результат имел форму (M, H). Который ...

0
Saad Hussain 5 Мар 2021 в 04:24

1 ответ

Лучший ответ

Это должно работать:

(torch.unsqueeze(a, 2)*b).sum(axis=0)
>>> tensor([[12, 14, 16],
            [30, 34, 38]])
2
user15270287 5 Мар 2021 в 02:48