torch.nn概述
nn是Neural Network的简称,帮助程序员方便执行如下的与神经网络相关的行为:
(1)创建神经网络
(2)训练神经网络
(3)保存神经网络
(4)恢复神经网络
torch.nn是专门为神经网路设计的模块化接口
nn构建与autograd之上,可以用来定义和运行神经网络
nn.Parameter 实际上也是tensor 是一个多维矩阵,是Varible类中的一个特殊类
nn.Linear 全连接层
nn.functional 定义了一些创建神经网络所需要的常见函数
nn.Module 它是一个抽象概念,既可以表示神经网络中的某个层(layer) , 也可以表示包含很多层的神经网络
nn.Sequential 可以构建一个串联的神经网络, 继承自nn.Module