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'>