GetByteSize
GetDataType
函数功能
获得Tensor的数据类型。
函数原型
MxBase::TensorDType GetDataType() const
返回参数说明
| 数据结构 | 说明 |
|---|---|
| MxBase::TensorDType | Tensor的数据类型,请参见TensorDType。 |
父主题: Tensor
GetMemoryType
函数功能
获得Tensor的内存类型。
函数原型
MemoryData::MemoryType GetMemoryType() const
返回参数说明
| 数据结构 | 说明 |
|---|---|
| MemoryData::MemoryType | MemoryType的数据类型,请参见MemoryData。 |
父主题: Tensor
GetByteSize
函数功能
获得Tensor数据内存占的字节量。
函数原型
size_t GetByteSize() const
返回参数说明
| 数据结构 | 说明 |
|---|---|
| size_t | Tensor数据内存占的字节量。 |
父主题: Tensor
GetDeviceId
函数功能
获得Tensor数据所在的芯片的编号。
函数原型
int32_t GetDeviceId() const
返回参数说明
| 数据结构 | 说明 |
|---|---|
| int32_t | Tensor数据所在的芯片的编号(-1代表在Host侧)。 |
父主题: Tensor
ToDevice
函数功能
将Tensor数据转移到Device侧。
- 如果原始的内存为用户通过构造函数传入,则该原始内存由用户自行管理释放。
- 若是通过TensorMalloc申请的内存,则是将数据转移到Device侧,原始内存被自动释放。
函数原型
APP_ERROR ToDevice(int32_t deviceId)
参数说明
| 参数名 | 输入/输出 | 说明 |
|---|---|---|
| deviceId | 输入 | 将Tensor转移到deviceId对应的设备。 “deviceId”需为有效的设备ID。 |
返回参数说明
| 数据结构 | 说明 |
|---|---|
| APP_ERROR | 程序执行返回的错误码,请参考APP_ERROR说明。 |
父主题: Tensor
ToHost
函数功能
将Tensor数据转移到Host侧。
- 如果原始的内存为用户通过构造函数传入,则该内存由用户自行管理释放。
- 若是通过TensorMalloc申请的内存,则是将数据转移到Host侧,原始内存被自动释放,无需用户管理。
函数原型
APP_ERROR ToHost()
返回参数说明
| 数据结构 | 说明 |
|---|---|
| APP_ERROR | 程序执行返回的错误码,请参考APP_ERROR说明。 |
父主题: Tensor
在线提单