channel_shuffle

dragon.vm.torch.channel_shuffle(
  input,
  dim=0,
  groups=1,
  out=None
)[source]

Shuffle channels between a given number of groups. [Zhang et.al, 2017].

Parameters:
  • input (dragon.vm.torch.Tensor) – The input tensor.
  • dim (int, optional, default=0) – The dimension of channels.
  • groups (int, optional, default=1) – The number of groups.
  • out (dragon.vm.torch.Tensor, optional) – The output tensor.
Returns:

dragon.vm.torch.Tensor – The output tensor.