Shape

Return the shape an array.

Abstract Signature:

Shape(a: Tensor)

PyTorch

API: torch.nn.attention.flex_attention.BlockMask.shape
Strategy: Direct Mapping
Official Docs ↗

JAX (Core)

API: jax.numpy.shape
Strategy: Direct Mapping
Official Docs ↗

NumPy

API: numpy.shape
Strategy: Direct Mapping
Official Docs ↗

Keras

API: keras.ops.shape
Strategy: Direct Mapping
Official Docs ↗

TensorFlow

API: tf.shape
Strategy: Direct Mapping
Official Docs ↗

Apple MLX

API: mlx.core.array.shape
Strategy: Direct Mapping
Official Docs ↗

Flax NNX

API: nnx.variablelib.Variable.shape
Strategy: Direct Mapping
Official Docs ↗