Function at_least_3d_device

pub fn at_least_3d_device(
    a: impl AsRef<Array>,
    stream: impl AsRef<Stream>,
) -> Result<Array>

Convert array to have at least three dimensions.

Params

• a: The input array.

Example

use mlx_rs::{Array, ops::*};

let x = Array::from_int(1);
let out = at_least_3d(&x);