我有2个2d的numpy数组,A的形状(i,j)和B (i,k),其中j >> k。我想定义一个新的三维数组C,使得C中的每个元素都是A中每个列与整个矩阵B的广播元素的乘积。换句话说,作为一个正常的python循环,我会这样做。for x in range(j):但是,在这种情况下,j是非常大的,如果我能够使用
考虑两个数组之间的矩阵乘积:A = np.random.rand(2,10,10)我认为以上是一个1乘2乘以2乘2的矢量矩阵产品,在A的10乘10,2,2和3维上广播,结果的检验C证实了这个直觉,np.allclose(C[i,j], A.T[i,j] @ B)代表了所有的i,j。(size 10 is different from 2)
所以就广播而言,10乘10乘2张量和2乘2矩阵与矩阵<e