tensorplay.nn.functional.pixel_shuffle

tensorplay.nn.functional.pixel_shuffle(input: Tensor, upscale_factor: int) → Tensor[source]

Rearranges elements in a tensor of shape (*, C x r^2, H, W) to a tensor of shape (*, C, H x r, W x r).

input[n, c*r^2 + i*r + j, h, w]``.