ArangeΒΆ

Returns a 1-D tensor of values from the interval [start, end) taken with common difference step.

Abstract Signature:

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

PyTorch

API: torch.arange
Strategy: Direct Mapping

JAX (Core)

API: jnp.arange
Strategy: Direct Mapping

NumPy

API: np.arange
Strategy: Direct Mapping

Keras

API: keras.ops.arange
Strategy: Direct Mapping

TensorFlow

API: tf.range
Strategy: Direct Mapping

Apple MLX

API: mx.arange
Strategy: Direct Mapping

Flax NNX

API: jnp.arange
Strategy: Direct Mapping

PaxML / Praxis

API: jnp.arange
Strategy: Direct Mapping