mindspore.dataset.Dataset.output_shapes

mindspore.dataset.Dataset.output_shapes(estimate=False)[源代码]

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

参数:
  • estimate (bool) - 如果 estimateFalse ,将返回数据集第一条数据的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]]