torch 的RNN LSTM GRU

首先创造初始值

torch 的RNN LSTM GRU_第1张图片torch 的RNN LSTM GRU_第2张图片

torch 的RNN LSTM GRU_第3张图片

 

这里画出图帮助理解

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

torch 的RNN LSTM GRU_第4张图片

 

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

你可能感兴趣的:(torch,torch,RNN,LSTM,GRU)