mmpretrain.models.utils.channel_shuffle¶
- mmpretrain.models.utils.channel_shuffle(x, groups)[源代码]¶
Channel Shuffle operation.
This function enables cross-group information flow for multiple groups convolution layers.
- 参数:
x (Tensor) – The input tensor.
groups (int) – The number of groups to divide the input tensor in the channel dimension.
- 返回:
The output tensor after channel shuffle operation.
- 返回类型:
Tensor