小白学Pytorch系列--Torch.nn API Shuffle Layers(16)

最新推荐文章于 2023-12-12 08:11:47 发布

发呆的比目鱼

最新推荐文章于 2023-12-12 08:11:47 发布

阅读量357

点赞数

分类专栏： PyTorch框架文章标签： pytorch python 深度学习

本文链接：https://blog.csdn.net/weixin_42486623/article/details/129773730

版权

PyTorch框架专栏收录该内容

52 篇文章 8 订阅

订阅专栏

小白学Pytorch系列–Torch.nn API Shuffle Layers(16)

方法	注释
nn.ChannelShuffle	将形状为 $(*, C, H, W)$ 的张量中的通道划分为g组，并将它们重新排列为 $C^g,g,H,W)$ ，同时保持原始张量形状。

nn.ChannelShuffle

>>> channel_shuffle = nn.ChannelShuffle(2)
>>> input = torch.randn(1, 4, 2, 2)
>>> print(input)
[[[[1, 2],
   [3, 4]],
  [[5, 6],
   [7, 8]],
  [[9, 10],
   [11, 12]],
  [[13, 14],
   [15, 16]],
 ]]
>>> output = channel_shuffle(input)
>>> print(output)
[[[[1, 2],
   [3, 4]],
  [[9, 10],
   [11, 12]],
  [[5, 6],
   [7, 8]],
  [[13, 14],
   [15, 16]],
 ]]