INN.PixelShuffle1d

Jump to bottom

Zhang Yanbo edited this page Oct 27, 2022 · 2 revisions

CLASS `INN.PixelShuffle1d(r)`

Invertible version of 1-d pixel shuffle. It will turn [N, C, X*r] to shape [N, C*r, X], or vise versa.

r: size of pixel shuffle block size

Methods

`forward(x, log_p0=0, log_det_J=0)`

Return the pixel unshuffled x. If compute_p=True, logp and logdet will also returned.

`inverse(y, **args)`

Compute the pixel shuffled of y. The **args is a placeholder for consistent format.