PyTorch
API: torch.nn.attention.flex_attention.BlockMask.shape
Strategy: Direct Mapping
JAX (Core)
API: jax.numpy.shape
Strategy: Direct Mapping
NumPy
API: numpy.shape
Strategy: Direct Mapping
Keras
API: keras.ops.shape
Strategy: Direct Mapping
TensorFlow
API: tf.shape
Strategy: Direct Mapping
Apple MLX
API: mlx.core.array.shape
Strategy: Direct Mapping
Flax NNX
API: nnx.variablelib.Variable.shape
Strategy: Direct Mapping