mlx_rs::ops

Function expand_dims

Source 
pub fn expand_dims(a: impl AsRef<Array>, axis: i32) -> Result<Array>
Expand description

Similar to expand_dims_axes, but only takes a single axis.