将矩阵列乘以numpy中的矩阵

问题描述 投票:0回答:1

我正在尝试以更有效的方式计算此操作:

y = np.array([xi*M for xi in x.T])

其中x(n,m)矩阵,M(m,m)矩阵。输出y的形状为(n,m,m),每个y[i,:,:] = np.multiply(x[:,i],M)

我曾考虑使用einsum,但我很难理解如何使用下标。

python numpy matrix-multiplication
1个回答
0
投票
np.einsum('ik,jk->ijk', x.T, M)
© www.soinside.com 2019 - 2024. All rights reserved.