C/C++教程

Pytorch 第一个程序

本文主要是介绍Pytorch 第一个程序，对大家解决编程问题具有一定的参考价值，需要的程序猿们随着小编来一起学习吧！ 
import torch
from torch import nn
from torch.utils.data import DataLoader
from torchvision import datasets
from torchvision.transforms import ToTensor, Lambda, Compose
import matplotlib.pyplot as plt

1. DataLoader

是Pytorch用来加载数据的常用的类，返回一个可遍历的数据集对象

传入参数：

  • dataset (Dataset) – dataset from which to load the data.

  • batch_size (int, optional) – how many samples per batch to load (default: 1).

  • shuffle (bool, optional) – set to True to have the data reshuffled at every epoch (default: False)

2. torchvision

是一个包，里面包含了很多常用的视觉数据集

 

这篇关于Pytorch 第一个程序的文章就介绍到这儿，希望我们推荐的文章对大家有所帮助，也希望大家多多支持为之网！
您可能喜欢
栏目导航