It seems like PackedSequence Object can only be use by nn.RNN ,  nn.LSTM ,  nn.GRU.
How we can creates batches to use them in our own defined function/classes which in my case is mostly Attentions (dot product)
I would appreciate any help.