了解一维向量上的 np.matmul

Question

a = [1, 2, 3]
b = [10, 10, 10]

np.matmul(a, b) 结果为 60。

numpy如何将(3,)和(3,)维度相乘并返回点积而不是外积(3 * 3)或者抛出错误“维度不匹配”？

Answer 1

这直接来自numpy.matmul()的文档：

• 如果第一个参数是一维的，它被提升为矩阵 在其尺寸前加上 1。矩阵乘法后 前面的 1 被删除。
• 如果第二个参数是一维的，它被提升为矩阵 将 1 添加到其尺寸。矩阵乘法后 附加的 1 被删除。

---

因此，在矩阵乘法期间，输入 a 和 b 的形状分别转换为 (1, 3) 和 (3,1)。

根据矩阵乘法的规则，我们知道：

1 x 3   3 x 1
| |
--------   ===>   总结。

因此，我们得到一个标量的结果。