我有一个形状为(5，4，4)的ndarray，它是一个由5个矩阵4x4组成的集合.我想将ndarray乘以形状(3，5)的矩阵，并且我想得到形状(3，4，4)的ndarray，其中结果中的每个矩阵4x4是54×4矩阵与来自矩阵(3，5)的行的系数的线性组合.
我有以下非常基本的代码:
import numpy as np
np.random.seed(10)
X = np.random.rand(5,4,4)
A = np.random.rand(3,5)
print( A*X )
如我所料，这会生成错误消息"操作数无法与形状(3，5)(5，4，4)一起广播"
我读了一些关于广播的文章，但我似乎找不到一种方法来编写可以被Numpy矢量化的操作.
有谁有过类似的经历吗？