SetShape
GetSize
函数功能
用于获取张量数据对应内存大小。
size大小应与实际内存大小一致，否则可能会导致程序出现coredump情况。
函数原型
size_t GetSize() const;
返回参数说明
| 数据结构 | 说明 |
|---|---|
| size_t | 返回具体数值。 |
父主题： TensorBase
GetByteSize
函数功能
用于获取buffer字节数据量。
函数原型
size_t GetByteSize() const;
返回参数说明
| 数据结构 | 说明 |
|---|---|
| size_t | 返回具体数值。 |
父主题： TensorBase
SetShape
函数功能
设置TensorBase的形状。
函数原型
APP_ERROR SetShape(std::vector<uint32_t> shape);
参数说明
| 参数名 | 输入/输出 | 说明 |
|---|---|---|
| shape | 输入 | shape向量中各维度要求为正整数且单个或各项乘积需小于536,870,912（512 * 1024 * 1024），否则函数将抛出异常。 |
返回参数说明
| 数据结构 | 说明 |
|---|---|
| APP_ERROR | 程序执行返回的错误码，请参考APP_ERROR说明。 |
父主题： TensorBase
GetShape
函数功能
用于获取Tensor的形状。
函数原型
std::vector<uint32_t> GetShape() const;
返回参数说明
| 数据结构 | 说明 |
|---|---|
| std::vector<uint32_t> | shape数组。 |
父主题： TensorBase
GetStrides
函数功能
用于获取Tensor的步长。
函数原型
std::vector<uint32_t> GetStrides() const;
返回参数说明
| 参数名 | 说明 |
|---|---|
| std::vector<uint32_t> | Tensor的步长数组。 |
父主题： TensorBase
GetDeviceId
函数功能
用于获取Tensor的设备号。
函数原型
int32_t GetDeviceId() const;
返回参数说明
| 数据结构 | 说明 |
|---|---|
| tensor | 返回设备号，是一个int类型。 |
父主题： TensorBase
在线提单