mindspore.dataset.Dataset.output_types
- Dataset.output_types()[source]
Get the types of output data.
- Returns
list, list of data types.
Examples
>>> 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')]