Unbind

Removes a tensor dimension. Returns a tuple of all slices along a given dimension.

Abstract Signature:

Unbind(input: Tensor, dim: int = 0)

PyTorchJAX (Core)KerasTensorFlow

PyTorch

API: torch.unbind

Strategy: Direct Mapping

Official Docs ↗

JAX (Core)

API: —

Strategy: Macro '[jnp.squeeze(x, axis={dim}) for x in jnp.split({input}, {input}.shape[{dim}], axis={dim})]'

Keras

API: keras.ops.unstack

Strategy: Direct Mapping

Official Docs ↗

TensorFlow

API: tf.unstack

Strategy: Direct Mapping

Official Docs ↗