get_worker_info

class paddle.io.get_worker_info [源代码]

获取 paddle.io.DataLoader 子进程信息的函数,用于 paddle.io.IterableDataset 中划分子进程数据。子进程信息包含以下字段:

num_workers: 子进程数。

id: 子进程逻辑序号,从0到 num_workers - 1

dataset: 各子进程中数据集实例。

示例代码见 paddle.io.IterableDataset