NativeChannelShuffle ==================== Native kernel level implementation of the channel_shuffle. **Abstract Signature:** ``NativeChannelShuffle(input: Tensor, groups: int)`` .. raw:: html

PyTorch

API: torch.native_channel_shuffle
Strategy: Direct Mapping

JAX (Core)

API: jax.numpy.reshape
Strategy: Plugin (channel_shuffle_macro)