Shortcuts

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

Read the Docs v: latest
Versions
latest
stable
mmcls-1.x
mmcls-0.x
dev
Downloads
epub
On Read the Docs
Project Home
Builds

Free document hosting provided by Read the Docs.