Hello, I want to modify the LSTM with a custom gate. I understand how to modify the LSTM cell. What I don’t understand is how would the back-propagation go through this modified cell, without me integrating it back with the pytorch LSTM layer? If it doesn’t, how do I integrate my changes into the original code?