python - 将 numpy 数组的每个元素相乘并求和

Question

我有两个 numpy 数组，X 和 y。 X 的大小为 m，y 的大小为 n。我需要将 y 的每个元素乘以 X 的每个元素，然后求和。

类似[sum(X[0]*y) sum(X[1]*y) sum(X[n]*y)]

这就是我的意思

np.sum(X[:, np.newaxis] * y, axis=1)

但是通常 X 和 y 确实很大，所以这样做

X[:, np.newaxis] * y

创建一个巨大的临时数组，它会炸毁东西。有没有更好的实现方法？

Answer 1

如果您将 y 的每个元素乘以 X 的每个元素，只需先将 X 的所有元素相乘，然后使用数组 y 乘以这个数字和总和:

num = X.prod()

(num * y).sum()

编辑:你指定的数组可以通过数组X乘以y的元素之和得到:

X * y.sum()