PyTorch笔记6-mini batch

本系列笔记为莫烦PyTorch视频教程笔记 github源码

概要

Torch 中提供了一种整理数据结构的好东西，叫做 DataLoader，可以用来包装自己的数据，进行批训练，而且批训练可以有多种途径

import torch
import torch.utils.data as Data

torch.manual_seed(1)     # reproducible

<torch._C.Generator at 0x106f673d8>

DataLoader

DataLoader 是 Torch 用来包装数据的工具，如将 numpy array 等数据形式转成 Tensor，然后放进该包装器中。使用 DataLoader 可以有效迭代数据。
Data loader. Combines a dataset and a sampler, and provides single- or multi-process iterators over the dataset.