给定 PyTorch A (a X m) 和 B (m X b) 中的 2 个张量 2-D，是否有任何有效的方法来获得张量 C (m X a X b)，其中C[i,:,:] = A[:,i] @ B[i,:]？
这里我举一个例子：
A = torch.FloatTensor([[1,2],[3,4]])
B = torch.FloatTensor([[1,2,3],[4,5,6]])
结果：
C = torch.FloatTensor([[[1,2,3],[3,6,9]],[[12,15,18],[16,20,24]]])
我使用 for 循环完成了它。然而，它非常低效。