Solved by ptrblck in post #2 
              
              
                You are passing x to torch.onnx.export, while it seems dummy_input should have been passed. 
Could you check it or is it a copy-paste error?