mindspore.dataset.Dataset.get_batch_size

Dataset.get_batch_size()

Return the size of batch.

Returns

int, the batch size of data.

Examples

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