什么是 PyTorch?

PyTorch 是基于以下两个目的而打造的python科学计算框架:

  • 无缝替换NumPy,并且通过利用GPU的算力来实现神经网络的加速。
  • 通过自动微分机制,来让神经网络的实现变得更加容易。

本次教程的目标:

  • 深入了解PyTorch的张量单元以及如何使用Pytorch来搭建神经网络。
  • 自己动手训练一个小型神经网络来实现图像的分类。

注意

确保已安装torchtorchvision包。

../_img/tensor_illustration_flat.png

张量

../_img/autodiff.png

torch.autograd的简要介绍

../_img/mnist1.png

神经网络简介

../_img/cifar101.png

自己动手训练一个图像分类器