onnx.numpy_helper

from_array(tensor[, name])	将数组转换为 TensorProto，包括 onnx._custom_element_types 中定义的受支持类型。
from_dict(dict_[, name])	将 Python 字典转换为映射定义。
from_list(lst[, name, dtype])	将列表转换为序列定义。
from_optional(opt[, name, dtype])	将 optional 值转换为 Optional 定义。
to_array(tensor[, base_dir])	将 tensor 定义对象转换为 numpy 数组。
to_dict(map_proto)	将映射定义转换为 Python 字典。
to_list(sequence)	将序列定义转换为 Python 列表。
to_optional(optional)	将 optional 定义转换为 Python optional 值。

array

onnx.numpy_helper.from_array(tensor: ndarray, name: str | None = None) → TensorProto

将数组转换为 TensorProto，包括 onnx._custom_element_types 中定义的受支持类型。

参数：

• tensor – 一个 numpy 数组。

• name – （可选）张量的名称。

返回：

转换后的张量定义。

返回类型：

TensorProto

onnx.numpy_helper.to_array(tensor: TensorProto, base_dir: str = '') → ndarray

将张量定义对象转换为 numpy 数组。支持 onnx._custom_element_types 中定义的类型。

参数：

• tensor – 一个 TensorProto 对象。

• base_dir – 如果存在外部张量，base_dir 可以帮助找到其路径

返回：

转换后的数组。

返回类型：

arr

由于 numpy 不支持 ONNX 中定义的所有类型（float 8 类型、bfloat16、int4、uint4、float4e2m1），这两个函数使用了 :mod:onnx._custom_element_types 中定义的自定义 dtype。

sequence

onnx.numpy_helper.to_list(sequence: SequenceProto) → list[Any]

将序列定义转换为 Python 列表。

参数：

sequence – 一个 SequenceProto 对象。

返回：

转换后的列表。

返回类型：

list

onnx.numpy_helper.from_list(lst: list[Any], name: str | None = None, dtype: int | None = None) → SequenceProto

将列表转换为序列定义。

参数：

• lst – 一个 Python 列表

• name – （可选）序列的名称。

• dtype – （可选）输入列表中的元素类型，用于在转换空列表时指定序列值。

返回：

转换后的序列定义。

返回类型：

SequenceProto

dictionary

onnx.numpy_helper.to_dict(map_proto: MapProto) → dict[Any, Any]

将映射定义转换为 Python 字典。

参数：

map_proto – 一个 MapProto 对象。

返回：

转换后的字典。

onnx.numpy_helper.from_dict(dict_: dict[Any, Any], name: str | None = None) → MapProto

将 Python 字典转换为映射定义。

参数：

• dict – Python 字典

• name – （可选）映射的名称。

返回：

转换后的映射定义。

返回类型：

MapProto

optional

onnx.numpy_helper.to_optional(optional: OptionalProto) → Any | None

将 optional 定义转换为 Python optional 值。

参数：

optional – 一个 OptionalProto 对象。

返回：

转换后的 optional 值。

返回类型：

opt

onnx.numpy_helper.from_optional(opt: Any | None, name: str | None = None, dtype: int | None = None) → OptionalProto

将 optional 值转换为 Optional 定义。

参数：

• opt – 一个 Python optional

• name – （可选）optional 的名称。

• dtype – （可选）输入中的元素类型，用于在转换空的 none 值时指定 optional 值。dtype 必须是有效的 OptionalProto.DataType 值

返回：

转换后的 optional 定义。

返回类型：

optional