GetDimSize
函数功能
获取shape的某一维度的长度。
函数原型
int64_t GetDimSize(int32_t index) const
参数说明
参数 |
输入/输出 |
说明 |
|---|---|---|
index |
输入 |
需要获取的维度的索引，有效范围为[0, GetDims() - 1]。 |
返回值说明
int64_t，shape的index维度的取值。
约束说明
无。
调用示例
int64_t dimSize = shape->GetDimSize(0);
父主题： TensorShape类