Shape

Constructs a shape tuple equivalent to torch.Size.

Abstract Signature:

Shape(iterable = ())

PyTorchJAX (Core)NumPyKerasTensorFlowApple MLXFlax NNXPaxML / Praxis

PyTorch

API: torch.Size

Strategy: Direct Mapping

Official Docs ↗

JAX (Core)

API: —

Strategy: Macro 'tuple({iterable})'

NumPy

API: numpy.ma.core.shape

Strategy: Macro 'tuple({iterable})'

Keras

API: keras.ops.shape

Strategy: Macro 'tuple({iterable})'

TensorFlow

API: tf.shape

Strategy: Macro 'tuple({iterable})'

Apple MLX

API: mlx.core.array.shape

Strategy: Macro 'tuple({iterable})'

Flax NNX

API: nnx.variablelib.Variable.shape

Strategy: Macro 'tuple({iterable})'

PaxML / Praxis

API: —

Strategy: Macro 'tuple({iterable})'