| nnf_pixel_shuffle {torch} | R Documentation |
Pixel_shuffle
Description
Rearranges elements in a tensor of shape (*, C \times r^2, H, W) to a
tensor of shape (*, C, H \times r, W \times r).
Usage
nnf_pixel_shuffle(input, upscale_factor)
Arguments
input |
(Tensor) the input tensor |
upscale_factor |
(int) factor to increase spatial resolution by |
[Package torch version 0.13.0 Index]