Movatterモバイル変換


[0]ホーム

URL:


ContentsMenuExpandLight modeDark modeAuto light/dark, in light modeAuto light/dark, in dark modeSkip to content
onnx-array-api 0.3.1 documentation
Logo
onnx-array-api 0.3.1 documentation

Contents

More

Back to top

Source code for onnx_array_api.light_api

fromtypingimportDict,OptionalfromonnximportModelProtofrom..annotationsimportdomainfrom.modelimportOnnxGraph,ProtoTypefrom.varimportVar,Vars
[docs]defstart(opset:Optional[int]=None,opsets:Optional[Dict[str,int]]=None,ir_version:Optional[int]=None,)->OnnxGraph:""" Starts an onnx model. :param opset: main opset version :param opsets: others opsets as a dictionary :param ir_version: specify the ir_version as well :return: an instance of :class:`onnx_array_api.light_api.OnnxGraph` A very simple model: .. runpython:: :showcode: from onnx_array_api.light_api import start onx = start().vin("X").Neg().rename("Y").vout().to_onnx() print(onx) Another with operator Add: .. runpython:: :showcode: from onnx_array_api.light_api import start onx = ( start() .vin("X") .vin("Y") .bring("X", "Y") .Add() .rename("Z") .vout() .to_onnx() ) print(onx) """returnOnnxGraph(opset=opset,opsets=opsets,ir_version=ir_version)
[docs]defg()->OnnxGraph:""" Starts a subgraph. :return: an instance of :class:`onnx_array_api.light_api.OnnxGraph` """returnOnnxGraph(proto_type=ProtoType.GRAPH)

[8]
ページ先頭

©2009-2025 Movatter.jp