更新时间:2021-03-18 GMT+08:00
分享

AISimpleTensor::SetBuffer

设置tensor数据地址。

函数格式

void SetBuffer(void *data, const int32_t size, bool isown=false);

参数说明

参数

说明

取值范围

data

数据地址。

-

size

数据长度。

说明:

单位为字节。size必须与数据实际大小一致。

-

isown

是否在tensor生命周期结束后,由tensor释放data地址的内存。

  • false:默认值,data地址的内存在tensor生命周期结束后,由用户进行free操作。
  • true:在tensor生命周期结束后tensor释放该内存,用户不能进行free操作,否则导致重复释放。

-

返回值

无。

分享:

    相关文档

    相关产品