mlx.core.vecdot

Contents

mlx.core.vecdot#

vecdot(a: array, b: array, /, *, axis: int = -1, stream: None | Stream | Device = None) → array#

Compute the vector dot product of two arrays along an axis.

Parameters:

  • a (array) – Input array

  • b (array) – Input array

  • axis (int, optional) – Axis over which to compute the dot product. Default: -1.

Returns:

The vector dot product.

Return type:

array

Contents