mlx_rs::ops

Function outer_device

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

Compute the outer product of two 1-D arrays, if the array’s passed are not 1-D a flatten op will be run beforehand.