函数：get_dataset_buffer

产品	是否支持
Atlas A3 训练系列产品/Atlas A3 推理系列产品	√
Atlas A2 训练系列产品/Atlas 800I A2 推理产品/A200I A2 Box 异构组件	√
Atlas 训练系列产品	√
Atlas 推理系列产品	√
Atlas 200I/500 A2 推理产品	√
Atlas 200/300/500 推理产品	√

获取aclmdlDataset中的第n个aclDataBuffer。

C函数原型

aclDataBuffer* aclmdlGetDatasetBuffer(const aclmdlDataset *dataset, size_t index)

python函数

data_buffer = acl.mdl.get_dataset_buffer(dataset, index)

参数名	说明
dataset	int，aclmdlDataset数据对象的指针地址。需提前调用acl.mdl.create_dataset接口创建aclmdlDataset类型的数据。
index	int，表明获取的是第几个aclDataBuffer。

参数名

说明

dataset

int，aclmdlDataset数据对象的指针地址。

需提前调用acl.mdl.create_dataset接口创建aclmdlDataset类型的数据。

index

int，表明获取的是第几个aclDataBuffer。

返回值	说明
data_buffer	int，aclDataBuffer数据对象指针地址。获取成功，返回aclDataBuffer的地址。获取失败返回空地址。

返回值

说明

data_buffer

int，aclDataBuffer数据对象指针地址。

父主题： aclmdlDataset