mindspore.dataset.Dataset.output_shapes
- mindspore.dataset.Dataset.output_shapes(estimate=False)[源代码]
获取数据集对象中每列数据的shape。
- 参数:
estimate (bool) - 如果 estimate 为
False
,将返回数据集第一条数据的shape。 否则将遍历整个数据集以获取数据集的真实shape信息,其中动态变化的维度将被标记为None(可用于动态shape数据集场景)。默认值:False
。
- 返回:
list,每列数据的shape列表。
样例:
>>> import mindspore.dataset as ds >>> import numpy as np >>> >>> def generator1(): ... for i in range(1, 100): ... yield np.ones((16, 83, 83)), np.array([i]) >>> >>> dataset = ds.GeneratorDataset(generator1, ["data1", "data2"]) >>> output_shapes = dataset.output_shapes() >>> print(output_shapes) [[16, 83, 83], [1]]