PyTorch_Notebook
1. Introduction
1.1 学习基础
使用FashionMNIST数据集训练一个神经网络,该神经网络预测输入图像是否属于以下类别之一:T恤/上衣,裤子,套头衫,连衣裙,外套,凉鞋,衬衫,运动鞋,包袋或踝靴。
1.1.1 运行程序方式
- 在云端:简单的入门方法,有专用的Colab链接,由托管环境执行程序;
- 在本地:安装PyTorch和TorchVersion,下载执行自己编写的程序。
1.2 Tensers-张量
Tenser是一种特殊的数据结构,与数据和矩阵非常相似。在PyTorch中,使用张量对模型的输入和输出以及模型的参数进行编码。
Tenser类似于NumPy的ndarray,张量不仅可以在GPU或其他硬件上运行。实际上张量和Numpy数组通常可以共享相同的基础内存
1.2.1 张量的初始化
张量可以直接根据数据创建可得,数据类型在Python语言中是自动推断的:
<