pytorch中的向量乘法

时间:2019-05-06 08:38:58

标签: vector pytorch torch

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

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

2 个答案:

答案 0 :(得分:0)

尝试:

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

答案 1 :(得分:0)

我尝试了这个,它正在起作用:

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

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