set_release接口
接口功能
为输入张量对象设置是否需要释放数据内存标识。
C++函数原型
void SetRelease(bool releaseFlag)
Python函数
set_release(release_flag)
接口参数
参数 |
是否必选 |
说明 |
取值要求 |
|---|---|---|---|
release_flag |
是 |
是否需要释放数据内存的标识。 |
合法的bool类型。 |
使用样例
标识是否需要释放该内存。通常标识为False，标识该tensor由内部释放，不需要随infer tensor对象析构。
1 | infer_tensor.set_release(False) |
父主题： InferTensor