PixelUnshuffle ============== Reverses the PixelShuffle operation. **Abstract Signature:** ``PixelUnshuffle(input: Tensor, downscale_factor: int)`` .. raw:: html

PyTorch

API: torch.pixel_unshuffle
Strategy: Direct Mapping

JAX (Core)

API: jax.nn.pixel_unshuffle
Strategy: Direct Mapping

NumPy

API:
Strategy: Custom / Partial

Keras

API: tf.nn.space_to_depth
Strategy: Direct Mapping

TensorFlow

API: tf.nn.space_to_depth
Strategy: Direct Mapping

Apple MLX

API:
Strategy: Custom / Partial

Flax NNX

API: jax.nn.pixel_unshuffle
Strategy: Direct Mapping

PaxML / Praxis

API: jax.nn.pixel_unshuffle
Strategy: Direct Mapping