torch学习笔记(二) nn类结构-Linear
Linear 是module的子類,是參數(shù)化module的一種,與其名稱一樣,表示著一種線性變換。
創(chuàng)建
parent 的init函數(shù)
Linear的創(chuàng)建需要兩個參數(shù),inputSize 和 outputSize
inputSize:輸入節(jié)點數(shù)
outputSize:輸出節(jié)點數(shù)
所以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
版權聲明:本文為博主原創(chuàng)文章,轉載請附上博文鏈接!
總結
以上是生活随笔為你收集整理的torch学习笔记(二) nn类结构-Linear的全部內(nèi)容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: torch.nn.Embedding
- 下一篇: 机器翻译注意力机制及其PyTorch实现