[英]Python: Multiplying a list of vectors by a list of matrices as a single matrix operation
我有一个100个N-dimensional
向量的列表和一个100 MxN
矩阵的列表。 因此,您可以100xN
两个数据结构视为一个100xN
列表(或numpy数组)和一个100xMxN
列表(或numpy数组)。
我想做的是获取每个向量及其对应矩阵的点积,这样输出应为100 M-dimensional
矩阵(即100xM
列表或numpy数组)。
但是,我不确定如何执行此操作。 由于效率的明显原因,我不想重复进行此操作。 我也知道这不是基本的矩阵乘法。 我想我可能想使用np.einsum
,但是我对此不太熟悉。
有人愿意帮助吗?
您可以像这样使用np.einsum
np.einsum('ij,ikj->ik',a,b)
样品运行-
In [42]: M,N = 3,4
In [43]: a = np.random.rand(100,N)
In [44]: b = np.random.rand(100,M,N)
In [45]: np.einsum('ij,ikj->ik',a,b).shape
Out[45]: (100, 3)
您也可以使用np.matmul
或@
运算符(Python 3.x),尽管它似乎比einsum
-
np.matmul(a[:,None],b.swapaxes(1,2))[:,0]
声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.