diffusion_rs_common::nn::ops

Function pixel_shuffle

source 
pub fn pixel_shuffle(xs: &Tensor, upscale_factor: usize) -> Result<Tensor>