megengine.functional.nn.pixel_shuffle

pixel_shuffle(inp, upscale_factor)

将形状为 (…, C * r^2, H, W) 的张量的元素重排为一个形状为 (…, C, H * r, W * r) 的张量，其中 r 是一个放大系数， … 是零个或更多的 batch 维度。

参数:

• inp (Tensor) – 输入张量。

• upscale_factor (int) – pixel_shuffle 的上采样因子。

返回类型:

Tensor

返回:

输出张量。