bm_net_info_t
是 算能(Sophgo)BMRuntime C API 中用于描述加载的神经网络模型信息的核心结构体。它包含了模型的输入/输出张量的元数据(如形状、数据类型、内存占用等),是模型部署过程中配置输入输出、预分配内存和动态调整模型的关键依据。
结构体成员详解
以下是 bm_net_info_t
的主要成员及其作用:
成员变量 | 类型 | 描述 |
---|---|---|
input_num |
int |
模型的输入张量数量(例如分类模型通常为1个输入) |
output_num |
int |
模型的输出张量数量(例如目标检测模型可能有多个输出) |
input_dtypes |
bm_data_type_t* |
输入张量的数据类型数组(如 BM_FLOAT32 、BM_INT8< |