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