python - numpy 矩阵乘法简化 - 这可能吗？

Question

有没有办法简化

 a=np.dot(a,b)

就像你把 a=a+b 写成 a+=b 一样？ (a,b 都是np.array)

Answer 1

在 Python3.5+ 中你可以使用 @ operator用于矩阵乘法，例如:

import numpy as np

a = np.random.randn(4, 10)
b = np.random.randn(10, 5)

c = a @ b

这相当于调用 c = np.matmul(a, b)。尚不支持就地矩阵乘法 (@=)(并且在大多数情况下没有任何意义，因为输出通常与第一个输入具有不同的维度)。

另请注意 np.matmul (和 @)的行为与 np.dot 不同当一个或多个输入数组的维度大于 2 时(参见 here )。