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 ·
576 阅读 ·
1 评论