16 lines
495 B
ReStructuredText
16 lines
495 B
ReStructuredText
mindspore.dataset.BatchInfo
|
||
===========================
|
||
|
||
.. py:class:: mindspore.dataset.BatchInfo
|
||
|
||
此类提供了两种方法获取数据集的批处理数量（batch size）和迭代数（epoch），
|
||
用于 `batch` 操作中的输入参数 `batch_size` 和 `per_batch_map`。
|
||
|
||
.. py:method:: get_batch_num()
|
||
|
||
返回数据集的批处理数量（batch size）。
|
||
|
||
.. py:method:: get_epoch_num()
|
||
|
||
返回数据集的迭代数（epoch）。
|