Pytorch批量矩阵向量外积

我正在尝试使用 PyTorch 生成向量矩阵外积（张量）。假设向量v有尺寸p和矩阵M有尺寸qXr，乘积的结果应为 pXqXr。

Example:

#size: 2
v = [0, 1] 
#size: 2X3
M = [[0, 1, 2],
     [3, 4, 5]]
#size: 2X2X3
v*M = [[[0, 0, 0],
        [0, 0, 0]],
       [[0, 1, 2],
        [3, 4, 5]]]

对于两个向量v1 and v2， 我可以用torch.bmm(v1.view(1, -1, 1), v2.view(1, 1, -1))。这可以很容易地扩展到一批向量。但是，我无法找到矢量矩阵情况的解决方案。另外，我需要对批量向量和矩阵执行此操作。

您可以使用torch.einsum https://pytorch.org/docs/stable/torch.html#torch.einsum操作员：

torch.einsum('bp,bqr->bpqr', v, M) # batch-wise operation v.shape=(b,p) M.shape=(b,q,r)
torch.einsum('p,qr->pqr', v, M)    # cross-batch operation