torch 的RNN LSTM GRU

2023-05-07 01:19:22

首先創造初始值

torch 的RNN LSTM GRU

torch 的RNN LSTM GRU

torch 的RNN LSTM GRU

這裡畫出圖幫助了解

torch 的RNN LSTM GRU

其實對于每個單元來說 h就是輸出, 這裡可以看到最終兩者的結果是相同但是id 不同, 在tensorflow中兩者的id都是相同的

torch 的RNN LSTM GRU

真不明白torch建立的rnn單元為啥非得指明輸入的x的次元, 這樣的模型隻能接收一定的x,而tensorflow中rnn單元對輸入的次元是沒有限制的, 隻需指定内部的單元數量及可以了

torch RNN lstm gru

上一篇: torch:)——PyTorch: transforms用法詳解常見的transform操作

下一篇: tensorflow2------基礎API的使用

繼續閱讀