torch学习笔记(二) nn类结构-Linear
生活随笔
收集整理的這篇文章主要介紹了
torch学习笔记(二) nn类结构-Linear
小編覺得挺不錯的,現在分享給大家,幫大家做個參考.
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
版權聲明:本文為博主原創文章,轉載請附上博文鏈接!
總結
以上是生活随笔為你收集整理的torch学习笔记(二) nn类结构-Linear的全部內容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: torch.nn.Embedding
- 下一篇: 机器翻译注意力机制及其PyTorch实现