ray.data.from_numpy_refs

ray.data.from_numpy_refs(ndarrays: ObjectRef[numpy.ndarray] | List[ObjectRef[numpy.ndarray]]) → MaterializedDataset

从Ray对象引用列表创建一个 Dataset 到NumPy ndarrays。

示例

>>> import numpy as np
>>> import ray
>>> arr_ref = ray.put(np.array([1]))
>>> ray.data.from_numpy_refs(arr_ref)
MaterializedDataset(num_blocks=1, num_rows=1, schema={data: int64})

从NumPy数组引用的列表创建一个Ray数据集。

>>> ray.data.from_numpy_refs([arr_ref, arr_ref])
MaterializedDataset(num_blocks=2, num_rows=2, schema={data: int64})

参数:

ndarrays – 一个指向 NumPy ndarray 的 Ray 对象引用，或一个指向 NumPy ndarrays 的 Ray 对象引用的列表。

返回:

Dataset 持有来自给定 ndarrays 的数据。

开发者API: 此API可能会在Ray的次要版本之间发生变化。