python – Numpy Dot两个2-d数组的产品,在numpy中获得3-d数组

5b51 2022/1/14 8:23:27 python 字数 892 阅读 587 来源 www.jb51.cc/python

对不起,标题写得很糟糕.我试图并行化我的代码的一部分,并陷入一个点产品.我正在寻找一种有效的方法来做下面的代码,我确信有一个简单的线性代数解决方案,但我很困惑： puy = np.arange(8).reshape(2,4) puy2 = np.arange(12).reshape(3,4) print puy, '\n' print puy2.T zz = np.zeros([4,2,3])

概述

puy = np.arange(8).reshape(2,4)
puy2 = np.arange(12).reshape(3,4)

print puy,'\n'
print puy2.T

zz = np.zeros([4,2,3])

for i in range(4):
    zz[i,:,:] = np.dot(np.array([puy[:,i]]).T,np.array([puy2.T[i,:]]))

>>> np.einsum('ik,jk->kij',puy,puy2)
array([[[ 0,0],[ 0,16,32]],[[ 1,5,9],[ 5,25,45]],[[ 4,12,20],[12,36,60]],[[ 9,21,33],[21,49,77]]])
>>> np.allclose(np.einsum('ik,puy2),zz)
True

总结

以上是编程之家为你收集整理的python – Numpy Dot两个2-d数组的产品,在numpy中获得3-d数组全部内容，希望文章能够帮你解决python – Numpy Dot两个2-d数组的产品,在numpy中获得3-d数组所遇到的程序开发问题。

如果您也喜欢它,动动您的小指点个赞吧

除非注明，文章均由 laddyq.com 整理发布，欢迎转载。

转载请注明：
链接：http://laddyq.com
来源：laddyq.com
著作权归作者所有。商业转载请联系作者获得授权，非商业转载请注明出处。

python – Numpy Dot两个2-d数组的产品,在numpy中获得3-d数组

概述

总结

分类汇总

您的鼓励是对我最大的支持