diffusion_rs_common::nn::ops

Function pixel_unshuffle

source 
pub fn pixel_unshuffle(xs: &Tensor, downscale_factor: usize) -> Result<Tensor>