onnxscript_relu_function_definition_and_model_export.py

1import onnxscript
2from onnxscript.onnx_opset import ai_onnx_v18 as op
3
4# Define an ONNX Script function
5@onnxscript.script()
6def relu(x):
7    return op.Where(x > 0, x, 0)
8
9# The function can be called directly like a Python function
10import numpy as np
11input_data = np.array([-1, 0, 1], dtype=np.float32)
12output = relu(input_data)
13print(f"Output: {output}")
14
15# The function can also be converted to an ONNX ModelProto
16model = relu.to_model_proto()
17
18# Save the model
19import onnx
20onnx.save(model, "relu.onnx")