megengine.functional.nn.pixel_shuffle

pixel_shuffle(inp, upscale_factor)[源代码]

将形状为（, C x r^2, H, W）的向量中的元素重新排列为形状为（, C, H x r, W x r） 的向量，其中 r 是一个上采样因子，其中*是零或 batch 的维度。

参数

  • inp (Tensor) – 输入张量。

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

返回类型

Tensor

返回

输出张量。

上一页

megengine.functional.nn.pad

下一页

megengine.functional.nn.interpolate