numpy 3D阵列乘以2D阵列

Question

我有一个W = shape(D,C,F)的3D数组和一个X = shape(D,F)的2D数组。我想输出形状为(D,C)的矩阵。

当前，我要做的是迭代X和W中的行

for row in X.shape[0]:
    X[row].dot(W[row].T)

有什么不需要做迭代的方法吗？我可以像X.dot(W.T)一样吗？谢谢。

Answer 1

尝试在您的语句中为W.T编制索引。如果没有更多细节，我很难看清，但是像X.dot(W.T[:1])这样的东西似乎可以在您的情况下起作用。