本文简要介绍python语言中 torch.nn.PixelShuffle
的用法。
用法:
class torch.nn.PixelShuffle(upscale_factor)
upscale_factor(int) -增加空间分辨率的因子
将形状为 的张量中的元素重新排列为形状为 的张量,其中 r 是一个放大因子。
这对于以 的步幅实现高效的 sub-pixel 卷积很有用。
见论文:使用高效 Sub-Pixel 卷积神经网络的实时单图像和视频超分辨率通过石等。 al (2016) 了解更多详情。
- 形状:
输入: ,其中 * 是零个或多个批次维度
输出: ,其中
例子:
>>> pixel_shuffle = nn.PixelShuffle(3) >>> input = torch.randn(1, 9, 4, 4) >>> output = pixel_shuffle(input) >>> print(output.size()) torch.Size([1, 1, 12, 12])
参数:
相关用法
- Python PyTorch PixelUnshuffle用法及代码示例
- Python PyTorch Pipe用法及代码示例
- Python PyTorch PitchShift用法及代码示例
- Python PyTorch PackageImporter.id用法及代码示例
- Python PyTorch Proxy用法及代码示例
- Python PyTorch PoissonNLLLoss用法及代码示例
- Python PyTorch ParagraphAggregator用法及代码示例
- Python PyTorch PooledEmbeddingsAllToAll用法及代码示例
- Python PyTorch PackageExporter.register_extern_hook用法及代码示例
- Python PyTorch Poisson用法及代码示例
- Python PyTorch Perceptron用法及代码示例
- Python PyTorch ParallelReadConcat用法及代码示例
- Python PyTorch ParameterList用法及代码示例
- Python PyTorch PairwiseDistance用法及代码示例
- Python PyTorch ParquetDataFrameLoader用法及代码示例
- Python PyTorch PackageExporter.register_intern_hook用法及代码示例
- Python PyTorch ParameterDict用法及代码示例
- Python PyTorch Pareto用法及代码示例
- Python PyTorch PackageExporter.close用法及代码示例
- Python PyTorch PooledEmbeddingsReduceScatter用法及代码示例
- Python PyTorch PackageExporter.register_mock_hook用法及代码示例
- Python PyTorch PReLU用法及代码示例
- Python PyTorch frexp用法及代码示例
- Python PyTorch jvp用法及代码示例
- Python PyTorch cholesky用法及代码示例
注:本文由纯净天空筛选整理自pytorch.org大神的英文原创作品 torch.nn.PixelShuffle。非经特殊声明,原始代码版权归原作者所有,本译文未经允许或授权,请勿转载或复制。