pytorch中的向量乘法

Question

在火炬中。我想将矩阵A的每个向量乘以矩阵B的每个向量：

A      = M x N
B      = L x N
result = (M x L) x N

Answer 1

尝试：

result = A[:, None, :] * B[None, ...]

Answer 2

我尝试了这个，它正在起作用：

torch.einsum('bj,aj->baj', input_unfolded, self.weights)

您可以使用这种方法设计任何乘法模式。