我得到了一个尺寸为(nx2x2)的矩阵,所以设了(3x2x2)的矩阵M:
[[[ 1. 5.]
[ 2. 4.]]
[[ 5. 25.]
[10. 20.]]
[[ 5. 25.]
[10. 20.]]]
我想做每个 2x2 矩阵的乘积运算符(点积)。换句话说,我想做以下事情:
其中 Mj 与M[j,:,:]python 中的相同。最简单的方法是使用如下所示的for循环:
prod=np.identity(2)
for temp in M:
prod=np.dot(prod,temp)
但是我不知道是否有矢量化的方法(我敢肯定这可能是一个重复的问题,但我无法找出适合Google的问题)
相关分类