keras(return_sequences)和pytorch(output, (h_n, c_n))的记录
近期需要将一份Keras代码转写为Pytorch,有关LSTM的东西比较麻烦,特此记录。
keras代码和介绍
pytorch LSTM的直观图
第二个链接国内可能wall,故此贴上Pytorch的理解图。
--------------------------------------------------------------------------------------------------------------------------------------------------
原创
2020-07-17 00:28:02 ·
1819 阅读 ·
0 评论