mlx_rs::ops

Function at_least_2d

Source 
pub fn at_least_2d(a: impl AsRef<Array>) -> Result<Array>
Expand description

Convert array to have at least two dimensions.

§Params

  • a: The input array.

§Example

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

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