TensorPlay
API symbolsnn
Copy
View MarkdownDownload .md

tensorplay.nn.functional.pixel_shuffle

tensorplay.nn.functional.pixel_shuffle(input: TensorBase, upscale_factor: int) TensorBase[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).

Port of aten::pixel_shuffle: output[n, c, h*r+i, w*r+j] = input[n, c*r^2 + i*r + j, h, w].

Ask DeepWiki