error happened when torch export onnx

[cwy0110]

torch export onnx :runtimeerror:the model cannot export to onnx, exception is : symbolic for dense_to_jagged produced an incorrect number of outputs(expected 2,but got 1)
So how to solve it? Thanks