Как умножить построчно на скаляр в pytorch? - PullRequest
0 голосов
/ 31 декабря 2018

Когда у меня есть тензор m формы [12, 10] и вектор s скаляров с формой [12], как я могу умножить каждый ряд m на соответствующий скаляр в s?

1 Ответ

0 голосов
/ 31 декабря 2018

Вам необходимо добавить соответствующее одноэлементное измерение:

m * s[:, None]

s[:, None] имеет размер (12, 1) при умножении тензора (12, 10) на (12, 1) тензор, который Pytoch знает для трансляции s вдоль второго одноэлементного измерения и правильное выполнение поэлементного произведения.

...