PyTorch
API: torch.is_same_size
Strategy: Direct Mapping
JAX (Core)
API: —
Strategy: Macro '{input}.shape == {other}.shape'
NumPy
API: —
Strategy: Macro '{input}.shape == {other}.shape'
Keras
API: —
Strategy: Macro 'keras.ops.shape({input}) == keras.ops.shape({other})'
TensorFlow
API: —
Strategy: Macro '{input}.shape == {other}.shape'
Apple MLX
API: —
Strategy: Macro '{input}.shape == {other}.shape'
Flax NNX
API: —
Strategy: Macro '{input}.shape == {other}.shape'
PaxML / Praxis
API: —
Strategy: Macro '{input}.shape == {other}.shape'