yobx.translate.translator#

class yobx.translate.translator.Translator(proto: ModelProto | FunctionProto | GraphProto, emitter: InnerEmitter | None = None)[source]#

Translates an ONNX graph into Python code using the selected emitter.

export(as_str: bool, single_line: bool = False) → str | List[str][source]#

Exports into a code.

Parameters:

  • as_str – as a single string or by rows

  • single_line – tries to compress the output into a single line

Returns:

list of instructions

extract_attributes(node: NodeProto) → Dict[str, Tuple[AttributeProto, Any]][source]#

Extracts all attributes of a node.

Parameters:

node – node proto

Returns:

dictionary