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

返回

输出张量。

上一页

megengine.functional.nn.pad

下一页

megengine.functional.nn.interpolate