mindspore.dataset.Dataset.create_dict_iterator
- mindspore.dataset.Dataset.create_dict_iterator(num_epochs=- 1, output_numpy=False, do_copy=True)[源代码]
基于数据集对象创建迭代器。输出的数据为字典类型。
- 参数:
num_epochs (int, 可选) - 迭代器可以迭代的最大次数。默认值:
-1
,迭代器可以迭代无限次。output_numpy (bool, 可选) - 输出的数据是否转为NumPy类型。如果为
False
,迭代器输出的每列数据类型为mindspore.Tensor
,否则为NumPy。默认值:False
。do_copy (bool, 可选) - 当参数 output_numpy 为
False
,即输出数据类型为mindspore.Tensor
时,可以将此参数指定为False
以减少拷贝,获得更好的性能。默认值:True
。
- 返回:
DictIterator,基于数据集对象创建的字典迭代器。
样例:
>>> import mindspore.dataset as ds >>> dataset = ds.GeneratorDataset([i for i in range(10)], "column1") >>> iterator = dataset.create_dict_iterator() >>> for item in iterator: ... # item is a dict ... print(type(item)) ... break <class 'dict'>