mlx_rs::ops

Function zeros_dtype_device

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

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