pytorch学习笔记

官方文档链接

pytorch提供了一种数据结构tensor可以和numpy相互转化。

variable:网络中的节点数据,在网络中的节点可以自动求梯度。(当前版本好像和tensor合并了,也就是说不用variable了。)

pytorch搭建的神经网络只处理二维的数据结构,所以要把一维的数组用转换成二维。

pytorch搭建神经网络要定义一个类,并继承torch.nn.module。类里面要定义两个函数:一个是__init__函数,功能是初始化,要继承父类的__init__函数。另一个是forward函数,用于组合前向传播网络的结构。