如何使用pytorch简化此乘法？

Question

我想像这样将两个矩阵相乘（标量乘法）。 但是这种方法看起来很愚蠢。我该如何简化呢？

A.size # B * S * V
B.size # B

A * B.unsqueeze(1).unsqueeze(2).expand_as(A)