当前位置: 首页>>代码示例 >>用法及示例精选 >>正文


Python PyTorch pixel_unshuffle用法及代码示例


本文简要介绍python语言中 torch.nn.functional.pixel_unshuffle 的用法。

用法:

torch.nn.functional.pixel_unshuffle(input, downscale_factor) → Tensor

参数

  • input(Tensor) -输入张量

  • downscale_factor(int) -增加空间分辨率的因子

通过将形状为 的张量中的元素重新排列为形状为 的张量来反转 PixelShuffle 操作,其中 r 是 downscale_factor

有关详细信息,请参阅 PixelUnshuffle

例子:

>>> input = torch.randn(1, 1, 12, 12)
>>> output = torch.nn.functional.pixel_unshuffle(input, 3)
>>> print(output.size())
torch.Size([1, 9, 4, 4])

相关用法


注:本文由纯净天空筛选整理自pytorch.org大神的英文原创作品 torch.nn.functional.pixel_unshuffle。非经特殊声明,原始代码版权归原作者所有,本译文未经允许或授权,请勿转载或复制。