PixelShuffle 差异对比

torch.nn.PixelShuffle

torch.nn.PixelShuffle(
   upscale_factor
)

更多请查看 torch.nn.PixelShuffle.

megengine.module.PixelShuffle

megengine.module.PixelShuffle(
  upscale_factor
)

更多请查看 megengine.module.PixelShuffle.

参数无差异，功能、用法相同。