Pytorch学习笔记

一：pytorch基础功能

什么是Pytorch？

 训练神经网络的三个步骤：

 训练、验证、测试：

  加载数据：

 

 

互换维度（将0，1两个维度互换）：

 合并矩阵：

数据类型：

使用cpu还是GPU处理tensor：

 梯度计算：

 二：如何用Pytorch定义一个神经网络

激活函数：

 

 损失函数：

优化方法：

 lr=learning rate

 全部训练过程：

 

 测试过程：

测试的时候调整为测试模式，且关闭对梯度的计算

 存档与读档：