Как умножить построчно на скаляр в pytorch?

Question

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

Answer 1

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

m * s[:, None]

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