Shape类接口
该类接口在tensor.h中定义。
GetDimNum
函数原型
size_t GetDimNum() const;
功能说明
获取Shape的维度个数。
参数说明
无。
返回值
参数名 |
类型 |
描述(参数说明、取值范围等) |
---|---|---|
- |
size_t |
Tensor Shape的维度个数。 |
异常处理
无。
约束说明
无。
GetDim
函数原型
int64_t GetDim(size_t idx) const;
功能说明
获取Shape第idx维的长度。
参数说明
参数名 |
输入/输出 |
类型 |
描述 |
---|---|---|---|
idx |
输入 |
size_t |
维度索引,索引从0开始。 |
返回值
参数名 |
类型 |
描述(参数说明、取值范围等) |
---|---|---|
- |
int64_t |
第idx维的长度 |
异常处理
无
约束说明
无。
GetDims
函数原型
std::vector<int64_t> GetDims() const;
功能说明
获取Shape所有维度组成的向量。
参数说明
无。
返回值
参数名 |
类型 |
描述(参数说明、取值范围等) |
---|---|---|
- |
std::vector<int64_t> |
Shape的所有维度组成的向量。 |
异常处理
无。
约束说明
无。
SetDim
函数原型
graphStatus SetDim(size_t idx, int64_t value);
功能说明
将Shape中第idx维度的值设置为value。
参数说明
参数名 |
输入/输出 |
类型 |
描述 |
---|---|---|---|
idx |
输入 |
size_t |
Shape维度的索引,索引从0开始。 |
value |
输入 |
int64_t |
需设置的值。 |
返回值
参数名 |
类型 |
描述(参数说明、取值范围等) |
---|---|---|
- |
graphStatus |
设置成功返回GRAPH_SUCCESS,否则,返回GRAPH_FAILED。 |
异常处理
无。
约束说明
无。