[ 仅 paddle 参数更多 ]torch.nn.ChannelShuffle¶

torch.nn.ChannelShuffle ¶

          torch.nn.ChannelShuffle(groups)

         

          paddle.nn.ChannelShuffle(groups, data_format='NCHW', name=None)

         

其中 Paddle 相比 PyTorch 支持更多其他参数，具体如下：

PyTorch	PaddlePaddle	备注
groups	groups	表示要把通道分成的组数。
-	data_format	表示输入 Tensor 的数据格式， PyTorch 无此参数， Paddle 保持默认即可。