当前位置: 首页>>代码示例 >>用法及示例精选 >>正文

Python PyTorch random_split用法及代码示例


本文简要介绍python语言中 torch.utils.data.random_split 的用法。

用法:

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

参数：

  • dataset(Dataset) -要拆分的数据集

  • lengths(序列) -要产生的分割长度

  • generator(torch.Generator) -用于随机排列的生成器。

将数据集随机拆分为给定长度的非重叠新数据集。可选择修复生成器以获得可重现的结果，例如：

>>> random_split(range(10), [3, 7], generator=torch.Generator().manual_seed(42))

相关用法


注：本文由纯净天空筛选整理自pytorch.org大神的英文原创作品 torch.utils.data.random_split。非经特殊声明，原始代码版权归原作者所有，本译文未经允许或授权，请勿转载或复制。