channel_shuffle

dragon.channel_shuffle(
  inputs,
  axis=0,
  group=1,
  **kwargs
)[source]

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

Parameters:
  • inputs (dragon.Tensor) – The inputs.
  • axis (int, optional, default=0) – The axis of channels.
  • group (int, optional, default=1) – The number of groups.
Returns:

dragon.Tensor – The output tensor.