pytorch
self.att_lstm = nn.LSTMCell(1536, 512)
self.lang_lstm = nn.LSTMCell(1024, 512)
请问上面的如何转成同等的keras或者tensorflow
pytorch
self.att_lstm = nn.LSTMCell(1536, 512)
self.lang_lstm = nn.LSTMCell(1024, 512)
请问上面的如何转成同等的keras或者tensorflow
Keras有 LSTM https://blog.csdn.net/jiangpeng59/article/details/77646186
1536和512,应该一个是Unit一个是Timesteps