跳到主要内容

GetMemoryType

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::TensorDTypeTensor的数据类型，请参见TensorDType。

父主题： Tensor

GetMemoryType​

函数功能​

获得Tensor的内存类型。

函数原型​

MemoryData::MemoryType GetMemoryType() const

返回参数说明​

数据结构说明
MemoryData::MemoryTypeMemoryType的数据类型，请参见MemoryData。

父主题： Tensor

GetByteSize​

函数功能​

获得Tensor数据内存占的字节量。

函数原型​

size_t GetByteSize() const

返回参数说明​

数据结构说明
size_tTensor数据内存占的字节量。

父主题： Tensor

GetDeviceId​

函数功能​

获得Tensor数据所在的芯片的编号。

函数原型​

int32_t GetDeviceId() const

返回参数说明​

数据结构说明
int32_tTensor数据所在的芯片的编号（-1代表在Host侧）。

父主题： Tensor

ToDevice​

函数功能​

将Tensor数据转移到Device侧。

  • 如果原始的内存为用户通过构造函数传入，则该原始内存由用户自行管理释放。
  • 若是通过TensorMalloc申请的内存，则是将数据转移到Device侧，原始内存被自动释放。

函数原型​

APP_ERROR ToDevice(int32_t deviceId)

参数说明​

参数名输入/输出说明
deviceId输入将Tensor转移到deviceId对应的设备。 “deviceId”需为有效的设备ID。

返回参数说明​

数据结构说明
APP_ERROR程序执行返回的错误码，请参考APP_ERROR说明。

父主题： Tensor

在线提单