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