# generate with torch python - <<'EOF' import torch, torch.nn as nn import torch.onnx model = nn.Linear(3, 1) x = torch.randn(1, 3) torch.onnx.export(model, x, "model.onnx", input_names=["in"], output_names=["out"]) EOF