获取Host侧的Tensor的buffer,首先判断Tensor是否在Host侧,然后在其中调用GetBuffer函数实现获取buffer的功能。
1 2 | template<typename T, typename... Ix> APP_ERROR TensorBase::GetValue(T &value, Ix... index) const; |
参数名 |
输入/输出 |
说明 |
---|---|---|
value |
输出 |
获取到对应Tensor的buffer。 |
index |
输入 |
待获取Tensor的索引。 |
数据结构 |
说明 |
---|---|
APP_ERROR |
程序执行返回的错误码,请参考APP_ERROR说明。 |