昇腾社区首页
中文
注册

函数: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 推理产品

使用说明

C函数原型

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

Python函数

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

函数功能

获取aclmdlDataset中的第n个aclDataBuffer。

输入说明

dataset:int,aclmdlDataset数据对象的指针地址。

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

index:int,表明获取的是第几个aclDataBuffer。

返回值说明

data_buffer:int,aclDataBuffer数据对象指针地址。

  • 获取成功,返回aclDataBuffer的地址。
  • 获取失败返回空地址。

约束说明