當前位置: 首頁>>編程示例 >>用法及示例精選 >>正文

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。非經特殊聲明，原始代碼版權歸原作者所有，本譯文未經允許或授權，請勿轉載或複製。