mmagic.models.archs.downsample

Module Contents

Functions

pixel_unshuffle(→ torch.Tensor)

Down-sample by pixel unshuffle.

mmagic.models.archs.downsample.pixel_unshuffle(x: torch.Tensor, scale: int) → torch.Tensor

Down-sample by pixel unshuffle.

Parameters

• x (Tensor) – Input tensor.

• scale (int) – Scale factor.

Returns

Output tensor.

Return type

Tensor