Linear 是module的子类,是参数化module的一种,与其名称一样,表示着一种线性变换。
创建
parent 的init函数
Linear的创建需要两个参数,inputSize 和 outputSize
inputSize:输入节点数
outputSize:输出节点数
所以Linear 有7个字段:
weight : Tensor , outputSize ×× inputSize
bias: Tensor ,outputSize
gradWeight: Tensor , outputSize ×× inputSize
gradBias: Tensor ,outputSize
gradInput: Tensor
output: Tensor
_type: output:type()
例子
module = nn.Linear(10, 5)
1
Forward Pass
---------------------
作者:bubbleoooooo
来源:优快云
原文:https://blog.youkuaiyun.com/u012936765/article/details/52671156
版权声明:本文为博主原创文章,转载请附上博文链接!
Linear模块的创建与参数介绍
博客介绍了Linear模块,它是module的子类,代表线性变换。创建Linear需inputSize和outputSize两个参数,分别为输入和输出节点数。还介绍了Linear的7个字段,最后给出创建例子。
889

被折叠的 条评论
为什么被折叠?



