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
来源:CSDN
原文:https://blog.csdn.net/u012936765/article/details/52671156
版权声明:本文为博主原创文章,转载请附上博文链接!

相关文章:

  • 2022-12-23
  • 2021-12-30
  • 2021-05-11
  • 2022-02-07
  • 2017-12-07
  • 2021-08-11
  • 2021-08-04
猜你喜欢
  • 2022-12-23
  • 2021-11-29
  • 2021-07-20
  • 2021-07-14
  • 2022-12-23
  • 2022-12-23
  • 2021-04-10
相关资源
相似解决方案