mindspore.dataset.Dataset.get_dataset_size

mindspore.dataset.Dataset.get_dataset_size()

返回一个epoch中的batch数。

返回：

int，batch的数目。

样例：

>>> import mindspore.dataset as ds
>>> import numpy as np
>>>
>>> # A generator return 66 samples
>>> def generator1():
...     for i in range(66):
...         yield np.ones((16, 83, 83)), np.array([i])
>>>
>>> dataset = ds.GeneratorDataset(generator1, ["data1", "data2"])
>>> dataset_size = dataset.get_dataset_size()
>>> print(dataset_size)
66