用于获取Tensor指针。
void* TensorBase::GetBuffer() const;
数据结构 |
说明 |
|---|---|
void* |
指针的首地址，即buffer指针。 |
APP_ERROR TensorBase::GetBuffer(void *&ptr, const std::vector<uint32_t> &indices) const;
参数名 |
输入/输出 |
说明 |
|---|---|---|
ptr |
输出 |
获取到的buffer指针。 |
indices |
输入 |
传入的索引列表，待复制的源内存。 |
数据结构 |
说明 |
|---|---|
APP_ERROR |
程序执行返回的错误码，请参考APP_ERROR说明。 |
template<typename T, typename... Ix> APP_ERROR TensorBase::GetBuffer(T* &value, Ix... index) const;
参数名 |
输入/输出 |
说明 |
|---|---|---|
value |
输出 |
获取到的buffer。 |
index |
输入 |
传入的索引值。 |
数据结构 |
说明 |
|---|---|
APP_ERROR |
程序执行返回的错误码，请参考APP_ERROR说明。 |