What should I do if I want to add something in cell node in LSTM?

1

Just like below
image
The red one is what I want to add. But LSTMCell and LSTM in nn seems don’t support this modification. Should I have to realize the LSTM by myself?