//clika-runtime/io.clika.runtime/Ops/pixelShuffle

pixelShuffle

[common]
fun pixelShuffle(input: Tensor, upscaleFactor: Long, mode: PixelShuffleMode = PixelShuffleMode.CRD): Tensor

pixelShuffle(input: Tensor, upscaleFactor: Long, mode: PixelShuffleMode = PixelShuffleMode.CRD): the pixel_shuffle operator. Rearranges channels into space (depth-to-space): channels-last [N, H, W, C] becomes [N, H*r, W*r, C/r^2].