You can do:
import torch.onnx
import torch.onnx.utils
...
trace, out = torch.jit.get_trace_graph(model, args=(input,))
trace = torch.onnx.utils._optimize_graph(trace.graph(), torch.onnx.OperatorExportTypes.ONNX)

This will give you more human-readable trace with onnx ops. Though, _optimize_graph is internal function.