单独重量的Pytorch LSTM

Question

我想用LSTM代替固定深度K的前馈网络，但是使用LSTM的好处是我们可以更好地管理爆炸梯度。

我不知道如何使用pytorch默认LSTM LSTMCell来做到这一点。本质上，我正在寻找的功能是能够在每个时间步1 .. K使用不同的权重矩阵。