mindspore.dataset.Dataset.get_batch_size
- Dataset.get_batch_size()[source]
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