pytorch 对抗样本_《AI安全之对抗样本入门》—3.4 PyTorch

3.4　PyTorch

PyTorch是torch的Python版本，是由Facebook开源的神经网络框架。PyTorch虽然是深度学习框架中的后起之秀，但是发展极其迅猛。PyTorch提供了NumPy风格的Tensor操作，熟悉NumPy操作的用户非常容易上手。我们以解决经典的手写数字识别的问题为例，介绍PyTorch的基本使用方法，代码路径为：

https://github.com/duoergun0729/adversarial_examples/blob/master/code/2-pytorch.ipynb

1. 加载相关库

加载处理经典的手写数字识别问题相关的Python库：

import os

import torch

import torchvision

from torch.autograd import Variable

import torch.utils.data.dataloader as Data

2. 加载数据集

PyTorch中针对常见的数据集进行了封装，免去了用户手工下载的过程并简化了预处理的过程。这里需要特别指出的是，PyTorch中每个Tensor包括输入节点，并且都可以有自己的梯度值，因此训练数据集要设置为train=True，测试数据集要设置为train=False：

train_data = torchvision.datasets.MNIST(

'dataset/mnist-pytorch', train=True,

transform=torchvision.transforms.ToTensor(), download=True

)