GetDataTypeSize
GetDeviceId
函数功能
用于获取Tensor的设备号。
函数原型
int32_t GetDeviceId() const;
返回参数说明
| 数据结构 | 说明 |
|---|---|
| tensor | 返回设备号，是一个int类型。 |
父主题： TensorBase
GetDataType
函数功能
用于获取内存中张量的数据类型。
函数原型
TensorDataType GetDataType() const;
返回参数说明
| 数据结构 | 说明 |
|---|---|
| TensorDataType | 请参考TensorDataType枚举说明。 |
父主题： TensorBase
GetDataTypeSize
函数功能
用于获取Tensor的数据类型。根据TensorDataType的具体类型返回该类型字节长度。
函数原型
uint32_t GetDataTypeSize() const;
返回参数说明
| 数据结构 | 说明 |
|---|---|
| uint32_t | Tensor的数据类型对应的字节长度。 |
父主题： TensorBase
IsHost
函数功能
用于判断Tensor对象是否在Host侧。
函数原型
bool IsHost() const;
返回参数说明
| 数据结构 | 说明 |
|---|---|
| bool | - true：在Host侧。 - false：不在Host侧。 |
父主题： TensorBase
IsDevice
函数功能
用于判断Tensor对象是否在Device侧。
函数原型
bool IsDevice() const;
返回参数说明
| 数据结构 | 说明 |
|---|---|
| bool | - true：在Device侧。 - false：不在Device侧。 |
父主题： TensorBase
GetBuffer
函数功能
用于获取Tensor指针。
函数原型1
void* GetBuffer() const;
返回参数说明1
| 数据结构 | 说明 |
|---|---|
| void* | 指针的首地址，即buffer指针。 |
函数原型2
APP_ERROR GetBuffer(void *&ptr, const std::vector<uint32_t> &indices) const;
参数说明2
| 参数名 | 输入/输出 | 说明 |
|---|---|---|
| ptr | 输出 | 获取到的buffer指针。 |
| indices | 输入 | 传入的索引列表，待复制的源内存。 |
返回参数说明2
| 数据结构 | 说明 |
|---|---|
| APP_ERROR | 程序执行返回的错误码，请参考APP_ERROR说明。 |
函数原型3
template<typename T, typename... Ix>
APP_ERROR GetBuffer(T* &value, Ix... index) const;
参数说明3
| 参数名 | 输入/输出 | 说明 |
|---|---|---|
| value | 输出 | 获取到的buffer。 |
| index | 输入 | 传入的索引值。 |
返回参数说明3
| 数据结构 | 说明 |
|---|---|
| APP_ERROR | 程序执行返回的错误码，请参考APP_ERROR说明。 |
父主题： TensorBase
在线提单