函数：get_dataset_num_buffers

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

获取aclmdlDataset中aclDataBuffer的个数。

C函数原型

size_t aclmdlGetDatasetNumBuffers(const aclmdlDataset *dataset)

python函数

index = acl.mdl.get_dataset_num_buffers(dataset)

参数名	说明
dataset	int，aclmdlDataset数据对象的指针地址。需提前调用acl.mdl.create_dataset接口创建aclmdlDataset类型的数据。

参数名

说明

dataset

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

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

返回值	说明
index	int，aclDataBuffer数据类型的个数。

父主题： aclmdlDataset