首先创造初始值

torch 的RNN LSTM GRU_LSTM

torch 的RNN LSTM GRU_LSTM_02

torch 的RNN LSTM GRU_LSTM_03

 

这里画出图帮助理解

torch 的RNN LSTM GRU_GRU_04

其实对于每个单元来说 h就是输出, 这里可以看到最终两者的结果是相同但是id 不同, 在tensorflow中两者的id都是相同的

torch 的RNN LSTM GRU_GRU_05

 

真不明白torch创建的rnn单元为啥非得指明输入的x的维度, 这样的模型只能接收一定的x,而tensorflow中rnn单元对输入的维度是没有限制的, 只需指定内部的单元数量及可以了