GetDataType
SetShape
函数功能
设置Tensor的形状(Shape)。
函数原型
APP_ERROR SetShape(std::vector<uint32_t> shape);
参数说明
| 参数名 | 输入/输出 | 说明 |
|---|---|---|
| shape | 输入 | Tensor的形状。 shape所代表的元素个数需要与Tensor的原shape所代表的元素个数相同。shape向量中各维度要求为正整数且单个或各项乘积需小于536,870,912(512 * 1024 * 1024),否则函数将抛出异常。 |
返回参数说明
| 数据结构 | 说明 |
|---|---|
| APP_ERROR | 程序执行返回的错误码,请参考APP_ERROR说明。 |
父主题: Tensor
GetShape
函数功能
获得Tensor的shape数据。
函数原型
std::vector<uint32_t> GetShape() const
返回参数说明
| 数据结构 | 说明 |
|---|---|
| std::vector<uint32_t> | Tensor的shape数据。 |
父主题: Tensor
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
在线提单