set_buffer接口
接口功能
为输入张量对象设置数据，数据大小和是否需要释放内存的标识。
C++函数原型
void SetBuffer(const void* buffer, size_t tensorbyteSize, bool tensorNeedRelease)
Python函数
set_buffer(tensor_data, tensor_need_release)
接口参数
参数 |
是否必选 |
说明 |
取值要求 |
|---|---|---|---|
tesnor_data |
是 |
张量数据。 |
合法的满足python缓冲区协议的类型对象。 |
tensor_need_release |
是 |
是否需要释放内存的标识。 |
合法的bool类型。 |
使用样例
data = [0,1,2,3] infer_tensor.set_buffer(data, False)
父主题： InferTensor