mindspore.dataset.Dataset.get_batch_size

mindspore.dataset.Dataset.get_batch_size()

获得数据集对象定义的批处理大小，即一个批处理数据中包含的数据条数。

返回：

int，一个批处理数据中包含的数据条数。

样例：

>>> import mindspore.dataset as ds
>>> dataset = ds.GeneratorDataset([i for i in range(10)], "column1")
>>> dataset = dataset.batch(2)
>>> batch_size = dataset.get_batch_size()
>>> print(batch_size)
2