Трансляция умножения Pytorch

Question

У меня есть два тензора PyTorch A с формой [30, 11, 32, 64] и B с формой [30, 11, 89]. Моя цель состоит в том, чтобы создать тензор C с формой [30, 11, 89, 32, 64], в котором тензор A должен быть расширен и воспроизведен в третьем измерении, а затем умножен на тензор B в соответствующие измерения.

Как это можно сделать в PyTorch?

Answer 1

Вы можете использовать семантику вещания так же, как numpy. Один из способов достижения sh это

C = A.reshape(30, 11, 1, 32, 64) * B.reshape(30, 11, 89, 1, 1)