nnf_pixel_shuffle: Pixel_shuffle

View source: R/nnf-pixelshuffle.R

nnf_pixel_shuffleR 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


torch documentation built on Aug. 21, 2025, 5:50 p.m.