为输入张量对象设置数据，数据大小和是否需要释放内存的标识。
void SetBuffer(const void* buffer, size_t tensorbyteSize, bool tensorNeedRelease)
set_buffer(tensor_data, tensor_need_release)
参数 |
是否必选 |
说明 |
取值要求 |
|---|---|---|---|
tesnor_data |
是 |
张量数据。 |
合法的满足python缓冲区协议的类型对象。 |
tensor_need_release |
是 |
是否需要释放内存的标识。 |
合法的bool类型。 |
为输入tensor设置数据对象，并标识是否需要释放该内存。通常标识为False，标识该tensor由内部释放，不需要随infer tensor对象析构。
1 2 | data = [0,1,2,3] infer_tensor.set_buffer(data, False) |