Solved by ptrblck in post #2 
              
              
                Yes, if you are not planning to export the model via ONNX, you could use enforce_sorted=False and pass the unsorted padded sequence to this method.