ray.data.range
ray.data.range#
- ray.data.range(n: int, *, parallelism: int = - 1) ray.data.dataset.Dataset[int] [source]#
Create a dataset from a range of integers [0..n).
Examples
>>> import ray >>> ds = ray.data.range(10000) >>> ds Dataset(num_blocks=200, num_rows=10000, schema=<class 'int'>) >>> ds.map(lambda x: x * 2).take(4) [0, 2, 4, 6]
- Parameters
n – The upper bound of the range of integers.
parallelism – The amount of parallelism to use for the dataset. Parallelism may be limited by the number of items.
- Returns
Dataset holding the integers.
PublicAPI: This API is stable across Ray releases.