mlx.core.arange

Contents

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

Overloaded function.

  1. arange(start : Union[int, float], stop : Union[int, float], step : Union[None, int, float], dtype: Optional[Dtype] = None, *, stream: Union[None, Stream, Device] = None) -> array

    Generates ranges of numbers.

    Generate numbers in the half-open interval [start, stop) in increments of step.

    Args:

    start (float or int, optional): Starting value which defaults to 0. stop (float or int): Stopping value. 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 of start, stop, or step are float. Otherwise will default to int32.

    Returns:

    array: The range of values.

    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.

  2. arange(stop : Union[int, float], step : Union[None, int, float] = None, dtype: Optional[Dtype] = None, *, stream: Union[None, Stream, Device] = None) -> array

Contents