tensorplay.nn.functional.channel_shuffle

tensorplay.nn.functional.channel_shuffle(input: Tensor, groups: int) → Tensor[source]

Divide the channels in a tensor into g groups and rearrange them as in ShuffleNet: (*, C, H, W) -> (*, C, H, W) with channels interleaved across groups.