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