Torch.onnx.export of a model returning dict

Is it supported? Can I specify the mapping of output keys to output_names? Will specifying output_names retrieve the specified output keys from the output dict?

It seems that by chance I get the correct outputs out of onnxruntime session run, probably because of the fact that python3 dicts preserve key order by default, but it’s not very reliable.


Same question here. Does anyone have an answer?