mlx_rs::ops

Function ones_dtype_device

Source 
pub fn ones_dtype_device(
    shape: &[i32],
    dtype: Dtype,
    stream: impl AsRef<Stream>,
) -> Result<Array>
Expand description

Similar to Array::ones but with a specified dtype.