megengine.functional.nn.pixel_shuffle

pixel_shuffle(inp, upscale_factor)[source]

Rearranges elements in a tensor of shape (…, C * r^2, H, W) to a tensor of shape (…, C, H * r, W * r), where r is an upscale factor, where … is zero or more batch dimensions.

Parameters

  • inp (Tensor) – input tensor.

  • upscale_factor (int) – upscale factor of pixel_shuffle.

Return type

Tensor

Returns

output tensor.

previous

megengine.functional.nn.pad

next

megengine.functional.nn.interpolate