Original LSTM cell uses dropout that uses different mask at every time step which is ad-Hoc and it leads to unstable results.According this paper we should use same dropout masks at every time step. Variational RNN 6
Here is the screenshot what should ideally happen

Screenshot_2020-05-03 1512 05287 pdf704×270

Keras supports this with (dropout and recurrent dropout)
Is there any neat implementation for this pytorch? Thanks for Helping