TensorPlay
API symbolsnn
Copy
View MarkdownDownload .md

tensorplay.nn.functional.pixel_unshuffle

tensorplay.nn.functional.pixel_unshuffle(input: TensorBase, downscale_factor: int) → TensorBase[source]

Reverses the pixel_shuffle() transformation: (*, C, H x r, W x r) -> (*, C x r^2, H, W).

tensorplay.nn.functional.pixel_shuffle

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) .

tensorplay.nn.functional.poisson_nll_loss

Compute the Poisson negative log likelihood loss.

Ask DeepWiki