【算能】BMCV库中与张量处理相关的核心数据结构及其详细说明

以下是算能(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