set_buffer接口

接口功能

为输入张量对象设置数据，数据大小和是否需要释放内存的标识。

C++函数原型

void SetBuffer(const void* buffer, size_t tensorbyteSize, bool tensorNeedRelease)

Python函数

set_buffer(tensor_data, tensor_need_release)

接口参数

使用样例

为输入tensor设置数据对象，并标识是否需要释放该内存。通常标识为False，标识该tensor由内部释放，不需要随infer tensor对象析构。

1
2

data = [0,1,2,3]
infer_tensor.set_buffer(data, False)