元素矩阵向量乘法

您想执行矩阵乘法运算( __matmul__ )以批量方式。直观地说,您可以使用批处理matmul运算符 torch.bmm .请记住,你首先需要解一个维度上的问题 v 这样它就变成了一个三维张量。在本例中,使用 None 像 v[..., None] 将提供一个 (n, 3, 1) .

具有 手电筒bmm :

>>> torch.bmm(m, v[..., None])

事实证明, torch.matmul 开箱即用地处理此案例:

>>> torch.matmul(m, v[..., None]) # same as m@v[..., None]

如果你想明确控制操作,你可以选择 torch.einsum :

>>> torch.einsum('bij,bj->bi', m, v)