How to upgrade fastrnn's LayerNormLSTM to support PackedSequence?

jit
1

In fastrnn’s custom LSTM, we could conveniently define JIT LayerNormLSTM. I am wondering that how could we also make it support PackedSequence as input data for varied lengths of tensors ?

2

This should be supported: