channel_shuffle¶
dragon.vm.torch.nn.functional.
channel_shuffle
(
input,
groups
)[source]¶Apply group shuffle to each channel of input. [Zhang et.al, 2017].
- Parameters:
- input (dragon.vm.torch.Tensor) – The input tensor.
- groups (int) – The number of shuffle groups.
- Returns:
dragon.vm.torch.Tensor – The output tensor.
See also