torch 的RNN LSTM GRU

首先創造初始值   這裏畫出圖幫助理解 其實對於每個單元來說 h就是輸出, 這裏可以看到最終兩者的結果是相同但是id 不同, 在tensorflow中兩者的id都是相同的   真不明白torch創建的rnn單元爲啥非得指明輸入的x的維度, 這樣的模型只能接收一定的x,而tensorflow中rnn單元對輸入的維度是沒有限制的, 只需指定內部的單元數量及可以了
相關文章
相關標籤/搜索