详解onnx.numpy_helper.from_array()方法在Python中的使用及示例

onnx.numpy_helper.from_array()方法是onnx模块中的一个辅助函数，用于将NumPy数组转换为ONNX TensorProto类型的数据。

该方法的语法如下：

from_array(array, name=None)

其中，array表示输入的NumPy数组，name表示转换后的TensorProto的名称，默认为None。

使用该方法，可以将NumPy数组转换为ONNX TensorProto类型的数据，方便在使用ONNX模型时进行处理和传递。

下面是一个使用示例：

import numpy as np
from onnx import numpy_helper

array = np.array([[1, 2, 3], [4, 5, 6]])
tensorproto = numpy_helper.from_array(array, name='input_data')

print(tensorproto)

以上代码首先导入了numpy和numpy_helper模块。然后创建了一个2x3的NumPy数组。接着使用from_array()方法将该数组转换为TensorProto类型的数据，并指定名称为'input_data'。最后打印输出了转换后的TensorProto数据。

运行以上代码，输出结果如下：

dims: 2
dim {
  dim_value: 2
}
dim {
  dim_value: 3
}
data_type: 6
raw_data: "\001\000\000\000\000\000\000\000\002\000\000\000\000\000\000\000\003\000\000\000\000\000\000\000\004\000\000\000\000\000\000\000\005\000\000\000\000\000\000\000\006\000\000\000\000\000\000\000"
name: "input_data"

可以看到，转换后的TensorProto数据包含数组的维度、数据类型以及原始数据。

这样，在使用ONNX模型时，可以使用该方法将NumPy数组转换为TensorProto数据，并进行传递和处理。