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 数组，否则转换为 mindspore.Tensor 。默认值： False 。

• do_copy (bool, 可选) - 指定转换输出类型为 mindspore.Tensor 时是否拷贝数据，否则直接复用数据缓冲区以获得更好的性能，仅当 output_numpy 为 False 时有效。默认值： True 。

返回：

Iterator，返回字典形式样本的迭代器。

样例：

>>> import mindspore.dataset as ds
>>>
>>> dataset = ds.GeneratorDataset([i for i in range(10)], "data")
>>> num_epochs = 3
>>> iterator = dataset.create_dict_iterator(num_epochs=num_epochs)
>>> for epoch in range(num_epochs):
...     for item in iterator:
...         # output is of type dict
...         print(type(item))
...         break
...     break
<class 'dict'>