以下是算能(Sophgo)BMCV 库中与张量(Tensor)处理相关的核心 数据结构 及其详细说明,涵盖张量属性、内存布局和数据类型等关键定义,适用于深度学习模型推理和后处理任务。
一、核心张量数据结构
1. bm_tensor_t
- 功能:描述张量的元数据及设备内存信息,是BMCV张量操作的核心结构。
- 定义:
typedef struct { bm_shape_t shape; // 张量维度信息(NCHW/NHWC等格式) bm_data_type data_type; // 数据类型(如FP32/INT8) bm_device_mem device_mem; // 设备内存句柄(存储张量数据) bm_tensor_private* priv; // 内部私有数据(用户无需操作) } bm_tensor_t;
- 关键字段说明:
-
bm_shape_t
-