rnn与gru, lstm, bi-lstm代码差异
rnn 与gru区别
两者网络接口相同,只需要在网络定义里替换一下相互名字即可
self.gru = nn.GRU(input_size, hidden_size, num_layers, batch_first=True)
self.rnn = nn.RNN(input_size, hidden_size, num_layers, batch_first=True)
其它输入数据和输出接口保持不变
rnn与lstm区别
rnn与lstm网络接口定义一样,只是换接口名称,但是lstm前向增加了cell_s
原创
2021-05-10 11:36:53 ·
569 阅读 ·
1 评论