mlx.core.inner#

inner(a: array, b: array, /, *, stream: None | Stream | Device = None) → array#

对于一维数组是普通的内积，在更高维度上是对最后几个轴求和的积。

参数:

• a (array) – 输入数组

• b (array) – 输入数组

返回值:

内积。

返回类型:

array