pytorch搭建RNN-LSTM循環神經網絡[分類]詳解

由於RNN在長序列中進行反向傳播，極可能會出現梯度消失或者梯度爆炸的狀況，而LSTM能解決該問題，下面就使用LSTM對MNIST數據集進行分類操做python 這裏主要講解搭建RNN部分，其餘部分和前文中CNN搭建相似。web 可參考pytorch搭建CNN卷積神經網絡詳解 🌵 搭建LSTM 重要！： class LSTM_RNN(nn.Module): """搭建LSTM"""

>>阅读原文<<