Shape类
Shape构造函数和析构函数
函数功能
Shape构造函数和析构函数。
函数原型
Shape();
~Shape();
explicit Shape(const std::vector<int64_t>& dims)
参数说明
参数名 |
输入/输出 |
描述 |
---|---|---|
dims |
输入 |
Shape的取值内容。 Shape表征张量数据的维度大小,用std::vector<int64_t>表征每一个维度的具体大小。 |
返回值
Shape构造函数返回Shape类型的对象。
异常处理
无。
约束说明
无。
GetDim
函数功能
获取Shape第idx维的长度。
函数原型
int64_t GetDim(size_t idx) const
参数说明
参数名 |
输入/输出 |
描述 |
---|---|---|
idx |
输入 |
维度索引,索引从0开始。 |
返回值
参数名 |
类型 |
描述 |
---|---|---|
- |
int64_t |
第idx维的长度 |
异常处理
无
约束说明
无。
GetDims
函数功能
获取Shape所有维度组成的向量。
函数原型
std::vector<int64_t> GetDims() const
参数说明
无。
返回值
参数名 |
类型 |
描述 |
---|---|---|
- |
std::vector<int64_t> |
Shape的所有维度组成的向量。 |
异常处理
无。
约束说明
无。
GetDimNum
函数功能
获取Shape的维度个数。
函数原型
size_t GetDimNum() const
参数说明
无。
返回值
参数名 |
类型 |
描述 |
---|---|---|
- |
size_t |
Tensor Shape的维度个数。 |
异常处理
无。
约束说明
无。
GetShapeSize
函数功能
获取Shape中所有dim的累乘结果。
函数原型
int64_t GetShapeSize() const
参数说明
无。
返回值
参数名 |
类型 |
描述 |
---|---|---|
- |
int64_t |
返回所有dim的累乘结果。 |
异常处理
无。
约束说明
无。
SetDim
函数功能
将Shape中第idx维度的值设置为value。
函数原型
graphStatus SetDim(size_t idx, int64_t value)
参数说明
参数名 |
输入/输出 |
描述 |
---|---|---|
idx |
输入 |
Shape维度的索引,索引从0开始。 |
value |
输入 |
需设置的值。 |
返回值
参数名 |
类型 |
描述 |
---|---|---|
- |
graphStatus |
设置成功返回GRAPH_SUCCESS,否则,返回GRAPH_FAILED。 |
异常处理
无。
约束说明
使用SetDim接口前,只能使用Shape(const std::vector<int64_t>& dims)构造shape对象。如果使用Shape()构造shape对象,使用SetDim接口将返回失败。