Function unflatten_device

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

Unflatten an axis of an array to a shape.

Params

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