onnx.defs

Opset 版本

onnx.defs.onnx_opset_version() → int[source]

返回当前操作集（opset）版本，用于域“ai.onnx”。

运算符和函数模式

onnx.defs.has()

has_schema(*args, **kwargs) 重载函数。

  1. has_schema(op_type: str, domain: str = ‘’) -> bool

  2. has_schema(op_type: str, max_inclusive_version: int, domain: str = ‘’) -> bool

onnx.defs.get_schema(*args, **kwargs)

重载函数。

  1. get_schema(op_type: str, max_inclusive_version: int, domain: str = ‘’) -> onnx.onnx_cpp2py_export.defs.OpSchema

返回运算符op_type的模式，以及特定版本。

  1. get_schema(op_type: str, domain: str = ‘’) -> onnx.onnx_cpp2py_export.defs.OpSchema

返回运算符op_type的模式，以及特定版本。

onnx.defs.get_all_schemas() → list[onnx.onnx_cpp2py_export.defs.OpSchema]

返回所有现有运算符的模式，针对最新版本。

onnx.defs.get_all_schemas_with_history() → list[onnx.onnx_cpp2py_export.defs.OpSchema]

返回所有现有运算符及其所有版本的模式。

onnx.defs.get_function_ops() → list[OpSchema][source]

返回定义为函数的运算符。

onnx.defs.register_schema(schema: OpSchema) → None[source]

注册用户提供的OpSchema。

如果需要，此函数会扩展为提供的域扩展可用运算符集版本。

参数：

schema – 要注册的OpSchema。

onnx.defs.deregister_schema(op_type: str, version: int, domain: str) → None

注销指定的OpSchema。

类 OpSchema

class onnx.defs.OpSchema

运算符的模式。

class AttrType

成员

FLOAT

INT

STRING

TENSOR

GRAPH

FLOATS

INTS

STRINGS

TENSORS

GRAPHS

SPARSE_TENSOR

SPARSE_TENSORS

TYPE_PROTO

TYPE_PROTOS

FLOAT = <AttrType.FLOAT: 1>
FLOATS = <AttrType.FLOATS: 6>
GRAPH = <AttrType.GRAPH: 5>
GRAPHS = <AttrType.GRAPHS: 10>
INT = <AttrType.INT: 2>
INTS = <AttrType.INTS: 7>
SPARSE_TENSOR = <AttrType.SPARSE_TENSOR: 11>
SPARSE_TENSORS = <AttrType.SPARSE_TENSORS: 12>
STRING = <AttrType.STRING: 3>
STRINGS = <AttrType.STRINGS: 8>
TENSOR = <AttrType.TENSOR: 4>
TENSORS = <AttrType.TENSORS: 9>
TYPE_PROTO = <AttrType.TYPE_PROTO: 13>
TYPE_PROTOS = <AttrType.TYPE_PROTOS: 14>
property name
property value
class Attribute
property default_value
property description
property name
property required
property type
class DifferentiationCategory

成员

未知

可微

不可微

Differentiable = <DifferentiationCategory.Differentiable: 1>
NonDifferentiable = <DifferentiationCategory.NonDifferentiable: 2>
Unknown = <DifferentiationCategory.Unknown: 0>
property name
property value
class FormalParameter
property description
property differentiation_category
property is_homogeneous
property min_arity
property name
property option
property type_str
property types
class FormalParameterOption

成员

单一

Optional

可变

Optional = <FormalParameterOption.Optional: 1>
Single = <FormalParameterOption.Single: 0>
Variadic = <FormalParameterOption.Variadic: 2>
property name
property value
class SupportType

成员

通用

实验性

COMMON = <SupportType.COMMON: 0>
EXPERIMENTAL = <SupportType.EXPERIMENTAL: 1>
property name
property value
class TypeConstraintParam
property allowed_type_strs
property description
property type_param_str
property all_function_opset_versions
property attributes
property context_dependent_function_opset_versions
property deprecated
属性 doc
属性 domain
属性 file
属性 function_body
属性 function_opset_versions
get_context_dependent_function(self: onnx.onnx_cpp2py_export.defs.OpSchema, arg0: bytes, arg1: list[bytes]) → bytes
get_context_dependent_function_with_opset_version(self: onnx.onnx_cpp2py_export.defs.OpSchema, arg0: int, arg1: bytes, arg2: list[bytes]) → bytes
get_function_with_opset_version(self: onnx.onnx_cpp2py_export.defs.OpSchema, arg0: int) → bytes
属性 has_context_dependent_function
属性 has_data_propagation_function
属性 has_function
属性 has_type_and_shape_inference_function
属性 inputs
静态 is_infinite(arg0: int) → bool
属性 line
属性 max_input
属性 max_output
属性 min_input
属性 min_output
属性 name
属性 outputs
属性 since_version
属性 support_level
属性 type_constraints

异常

类 onnx.defs.SchemaError

常量

onnx 包中正式支持的域。

from onnx.defs import (
    ONNX_DOMAIN,
    ONNX_ML_DOMAIN,
    AI_ONNX_PREVIEW_TRAINING_DOMAIN,
)
print(f"ONNX_DOMAIN={ONNX_DOMAIN!r}")
print(f"ONNX_ML_DOMAIN={ONNX_ML_DOMAIN!r}")
print(f"AI_ONNX_PREVIEW_TRAINING_DOMAIN={AI_ONNX_PREVIEW_TRAINING_DOMAIN!r}")

ONNX_DOMAIN=''
ONNX_ML_DOMAIN='ai.onnx.ml'
AI_ONNX_PREVIEW_TRAINING_DOMAIN='ai.onnx.preview.training'