arange

rbc.stdlib.creation_functions.arange(start, stop=None, step=1, dtype=None, device=None)

Return evenly spaced values within a given interval.

Examples

>>> from rbc.stdlib import array_api
>>> @heavydb('int64[](int64)')
... def rbc_arange(start):
...     return array_api.arange(start)
>>> rbc_arange(5).execute()
array([0, 1, 2, 3, 4])

>>> @heavydb('double[](int64, float, float)')
... def rbc_arange3(start, stop, step):
...     return array_api.arange(start, stop, step)
>>> rbc_arange3(0, 1.0, 0.2).execute()
array([0. , 0.2, 0.4, 0.6, 0.8], dtype=float32)

>>> @heavydb('int8[](int64, int64)')
... def rbc_arange3(start, stop):
...     return array_api.arange(start, stop, dtype=array_api.int8)
>>> rbc_arange3(2, 5).execute()
array([2, 3, 4], dtype=int8)

Array-API ‘arange’ doc