Range

Returns a 1-D tensor of size floor((end - start) / step) + 1 with values from start to end with step step.

Abstract Signature:

Range(start: float = 0, end: float, step: float = 1)

PyTorchJAX (Core)NumPyKerasTensorFlowApple MLX

PyTorch

API: torch.range

Strategy: Direct Mapping

Official Docs ↗

JAX (Core)

API: jnp.arange

Strategy: Macro 'jnp.arange({start}, {end} + {step}, {step})'

NumPy

API: np.arange

Strategy: Macro 'np.arange({start}, {end} + {step}, {step})'

Keras

API: keras.ops.arange

Strategy: Macro 'keras.ops.arange({start}, {end} + {step}, {step})'

TensorFlow

API: tf.range

Strategy: Macro 'tf.range({start}, {end} + {step}, {step})'

Apple MLX

API: mlx.core.arange

Strategy: Macro 'mlx.core.arange({start}, {end} + {step}, {step})'