mlx.core.arange#
- arange(start: int | float, stop: int | float, step: None | int | float, dtype: Dtype | None = None, *, stream: None | Stream | Device = None) array #
- arange(stop: int | float, step: None | int | float = None, dtype: Dtype | None = None, *, stream: None | Stream | Device = None) array
Generates ranges of numbers.
Generate numbers in the half-open interval
[start, stop)
in increments ofstep
.- Parameters:
start (float or int, optional) – Starting value which defaults to
0
.step (float or int, optional) – Increment which defaults to
1
.dtype (Dtype, optional) – Specifies the data type of the output. If unspecified will default to
float32
if any ofstart
,stop
, orstep
arefloat
. Otherwise will default toint32
.
- Returns:
The range of values.
- Return type:
Note
Following the Numpy convention the actual increment used to generate numbers is
dtype(start + step) - dtype(start)
. This can lead to unexpected results for example if start + step is a fractional value and the dtype is integral.