标签： pytorch
我想像这样将两个矩阵相乘（标量乘法）。 但是这种方法看起来很愚蠢。我该如何简化呢？
A.size # B * S * V B.size # B A * B.unsqueeze(1).unsqueeze(2).expand_as(A)