onnx.numpy_helper¶
|
将数组转换为 TensorProto,包括 |
|
将 Python 字典转换为映射定义。 |
|
将列表转换为序列定义。 |
|
将 optional 值转换为 Optional 定义。 |
|
将 tensor 定义对象转换为 numpy 数组。 |
|
将映射定义转换为 Python 字典。 |
|
将序列定义转换为 Python 列表。 |
|
将 optional 定义转换为 Python optional 值。 |
array¶
- onnx.numpy_helper.from_array(tensor: ndarray, name: str | None = None) TensorProto [源代码]¶
将数组转换为 TensorProto,包括
onnx._custom_element_types
中定义的受支持类型。- 参数:
tensor – 一个 numpy 数组。
name – (可选)张量的名称。
- 返回:
转换后的张量定义。
- 返回类型:
- 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 对象。
- 返回:
转换后的列表。
- 返回类型:
dictionary¶
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