[ 参数完全一致 ]torch.utils.data.random_split

torch.utils.data.random_split

torch.utils.data.random_split(dataset,
                            lengths,
                            generator=<torch._C.Generator object>)

paddle.io.random_split

paddle.io.random_split(dataset,
                    lengths,
                    generator=None)

两者参数和用法完全一致,具体如下:

参数映射

| PyTorch | PaddlePaddle | 备注 | | ————- | ———— | —————————————————— | | dataset | dataset | 表示可迭代数据集。 | | lengths | lengths | 总和为原数组长度的,子集合长度数组。 | | generator | generator | 指定采样 data_source 的采样器。默认值为 None。 |