python - @ 用于 matmul/inner，什么用于外积？

Question

在 Python 中，@ 运算符传递给元素的 __matmul__ 属性。当实现一个与实际后端无关的方法时，这会派上用场。例如

def inner(x, y):
    return x @ y
    # same:
    # return x.__matmul__(y)

实现一个内积，对于 x，y 是 numpy 数组或任何其他奇特的数组类。

外部产品是否也有类似的 API？

Answer 1

@ 运算符被添加为 PEP 465对于 __matmul__。外部产品没有这样的东西(也没有 dunder 方法)。

事实上，一旦第一个数组被 reshape ，外积就是一个简单的乘法(*):

import numpy as np

a = np.array([1,2,3])
b = np.array([10, 100])

np.outer(a, b)

a[:,None] * b

两种产品的输出:

array([[ 10, 100],
       [ 20, 200],
       [ 30, 300]])