onnx2torch is an ONNX to PyTorch converter. Our converter:
Is easy to use – Convert the ONNX model with the function call convert;
Is easy to extend – Write your own custom layer in PyTorch and register it with @add_converter;
Convert back to ONNX – You can convert the model back to ONNX using the torch.onnx.export function.