torch学习笔记(二) nn类结构-Linear

Linear 是module的子类，是参数化module的一种，与其名称一样，表示着一种线性变换。

创建

parent 的init函数

Linear的创建需要两个参数，inputSize 和 outputSize
inputSize：输入节点数
outputSize：输出节点数
所以Linear 有7个字段:

• weight : Tensor ， outputSize $\times$ inputSize
• bias: Tensor ，outputSize
• gradWeight: Tensor ， outputSize $\times$ inputSize
• gradBias: Tensor ，outputSize
• gradInput: Tensor
• output: Tensor
• _type: output:type()

例子

module = nn.Linear(10, 5)

Forward Pass