mindspore.dataset.Dataset.output_types

查看源文件
mindspore.dataset.Dataset.output_types()[源代码]

获取数据集对象中每列数据的数据类型。

返回:

list,每列数据的数据类型列表。

样例:

>>> import mindspore.dataset as ds
>>> import numpy as np
>>>
>>> def generator1():
...     for i in range(1, 100):
...         yield np.ones((16, 83, 83)).astype(np.float32), np.array([i]).astype(np.int32)
>>>
>>> dataset = ds.GeneratorDataset(generator1, ["data1", "data2"])
>>> output_types = dataset.output_types()
>>> print(output_types)
[dtype('float32'), dtype('int32')]