mlx.core.inner

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

Ordinary inner product of vectors for 1-D arrays, in higher dimensions a sum product over the last axes.

参数:

• a (array) -- Input array

• b (array) -- Input array

返回:

The inner product.

返回类型:

array