n个矩阵的乘积算子没有循环

我得到了一个尺寸为(nx2x2)的矩阵,所以设了(3x2x2)的矩阵M:


[[[ 1.  5.]

  [ 2.  4.]]


 [[ 5. 25.]

  [10. 20.]]


 [[ 5. 25.]

  [10. 20.]]]

我想做每个 2x2 矩阵的乘积运算符(点积)。换句话说,我想做以下事情:

http://img3.mukewang.com/60b5f7650001d4d300510054.jpg

其中 Mj 与M[j,:,:]python 中的相同。最简单的方法是使用如下所示的for循环:


prod=np.identity(2)

for temp in M:

    prod=np.dot(prod,temp)

但是我不知道是否有矢量化的方法(我敢肯定这可能是一个重复的问题,但我无法找出适合Google的问题)


皈依舞
浏览 143回答 2
2回答
打开App,查看更多内容
随时随地看视频慕课网APP

相关分类

Python