Is there a rayon parallel implementation for creating array?

[arthurlee945]

Hello!

i'm currently creating image array dim of (1, 3, x, y) doing something like this.

    ndarray::Array::from_shape_fn(
        (1, 3, size.width as usize, size.height as usize),
        // BGR -> RGB
        |(_, c, x, y)| (bytes[3 * x + 3 * y * (size.width as usize) + (2 - c)] as f32) / 255., // u8::MAX
    )

i'm refactoring iteration and map using rayon feature right now and would love to do that for this ndarray creation step as well.
I'm struggling finding par implementation for ndarray creation step. If you guys can give some ideas that would be awesome.