PyTorch基础教程 PDF 下载_Java知识分享网-免费Java资源下载

Java知识分享网 - 轻松学习从此开始！

当前位置: 主页 > Java文档 > Python技术 >

PyTorch基础教程 PDF 下载

分享到：

时间:2024-07-31 14:18来源:http://www.java1234.com 作者:转载 侵权举报

PyTorch基础教程

失效链接处理

PyTorch基础教程 PDF 下载

转载自：http://www.python222.com/article/997

相关截图：

主要内容：

2. 加载和预处理数据

创建一个Python文件，比如叫 main.py ，然后开始编写代码。首先，导入必要的库：

import torch
import torchvision
import torchvision.transforms as transforms
from torch.utils.data import DataLoader
# 设置数据转换
transform = transforms.Compose([
transforms.ToTensor(), # 转换图像为Tensor
transforms.Normalize((0.5,), (0.5,)) # 标准化
])
# 下载训练集和测试集
train_set = torchvision.datasets.MNIST(root='./data', train=True, download=True,
transform=transform)
test_set = torchvision.datasets.MNIST(root='./data', train=False, download=True,
transform=transform)
# 加载数据
train_loader = DataLoader(train_set, batch_size=64, shuffle=True)
test_loader = DataLoader(test_set, batch_size=64, shuffle=False)