平坦化时将张量与矩阵相乘

Question

我有一个张量A（可以是任意维）和一个矩阵M。我想通过将张量展平以得到一个矩阵来将它们相乘。

在MATLAB上，我可以简单地做到这一点：

function B = multiplyflatten(A,M)
B = M*A(:,:)

为了进行测试，可以使用]运行上述功能>

A=rand(8,5,4,4)
B=rand(1,8)
如何使用numpy ndarrays在python上获得等效的东西？下面的块不会产生等效的东西

def multiplyflatten(A,M):
    B=np.matmul(M,A[:,:])
A(:,:)在MATLAB中将许多维数压缩为矩阵的第二维。在Python上可以轻松做到这一点吗？ 

我有一个张量A（可以是任意维）和一个矩阵M。我想通过将张量展平以得到一个矩阵来将它们相乘。在MATLAB上，我可以简单地做到这一点：函数B = multipliflatten（A，...

Answer 1

如果我理解[A(:,:)在MATLAB中将许多维数压缩到矩阵的第二维中]