mlx.core.linalg.cross

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

计算两个数组沿指定轴的叉积。

叉积适用于指定轴上大小为 2 或 3 的数组。如果大小为 2，则假定第三个值为零。

参数:

• a (array) – 输入数组。

• b (array) – 输入数组。

• axis (int, optional) – 计算叉积的轴。默认值: -1。

• stream (Stream, optional) – 流或设备。默认为 None，此时使用默认设备的默认流。

返回:

a 和 b 沿指定轴的叉积。

返回类型:

array