mlx_rs::linalg

Function cross_device

Source 
pub fn cross_device(
    a: impl AsRef<Array>,
    b: impl AsRef<Array>,
    axis: Option<i32>,
    stream: impl AsRef<Stream>,
) -> Result<Array>
Expand description

Compute the cross product of two arrays along a specified axis.

The cross product is defined for arrays with size 2 or 3 in the specified axis. If the size is 2 then the third value is assumed to be zero.