大家好,又见面了,我是你们的朋友全栈君。...pytorch-DataLoader的使用
import torch
import torch.utils.data as Data
# [1, 1, 1]相当于一句话的word embedding,...[4, 4, 4], [5, 5, 5,], [6, 6, 6],[7, 7, 7], [8, 8, 8,], [9, 9, 9], [10, 10, 10]])
# [1, 2, 3]分别是这三句话的标签...从数据集中采样样本的方法。
# num_workers:int,可选。加载数据时使用多少子进程。默认值为0,表示在主进程中加载数据。
# collate_fn:callable,可选。...True表示如果最后剩下不完全的batch,丢弃。False表示不丢弃。