Custom RNN with PackSequence

1

Hi everyone, is there an easy way to implement a custom RNN cell that can support input of PackSequence? As far as I can see, the forward function implemented by RNNBase calls _backend.RNN, while _backend.RNN only supports limited modes, like LSTM and GRU.

1 Like
2

I did a similar thing a while ago. I copied relevant logic from RNNBase to a new class. It was quite painful…

3

Thanks for your reply! Hope to see the further support soon.