在火炬中。我想将矩阵A的每个向量乘以矩阵B的每个向量:
A = M x N
B = L x N
result = (M x L) x N
答案 0 :(得分:0)
尝试:
result = A[:, None, :] * B[None, ...]
答案 1 :(得分:0)
我尝试了这个,它正在起作用:
torch.einsum('bj,aj->baj', input_unfolded, self.weights)
您可以使用这种方法设计任何乘法模式。