如何在pytorch中按行乘以标量？

Question

当我有一个m形状的张量[12, 10]和一个s带有形状的标量向量时[12]，如何将每行m与相应的标量相乘s？

Answer 1

您需要添加相应的单例尺寸：

m * s[:, None]

s[:, None](12, 1)当(12, 10)张量乘以张量时，的大小为(12, 1)。pytoch知道会沿第二个单例维度广播 s，并正确执行“按元素”乘积。