为输入张量对象设置数据,数据大小和是否需要释放内存的标识。
void SetBuffer(const void* buffer, size_t tensorbyteSize, bool tensorNeedRelease)
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)