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