Function unflatten

pub fn unflatten(
    a: impl AsRef<Array>,
    axis: i32,
    shape: &[i32],
) -> Result<Array>

Unflatten an axis of an array to a shape.

Params

• a: input array
• axis: axis to unflatten
• shape: shape to unflatten into