IsSameSize

Returns True if two tensors have the same size.

Abstract Signature:

IsSameSize(input: Tensor, other: Tensor)

PyTorchJAX (Core)NumPyKerasTensorFlowApple MLXFlax NNXPaxML / Praxis

PyTorch

API: torch.is_same_size

Strategy: Direct Mapping

Official Docs ↗

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'