mlx_rs::ops

Function ones_like_device

Source
pub fn ones_like_device(
    input: impl AsRef<Array>,
    stream: impl AsRef<Stream>,
) -> Result<Array>
Expand description

An array of ones like the input.