GetDimSizes

函数功能

获取Tensor的shape的值，即Tensor的数据维度大小，用std::vector<int64_t>表征每一个维度的具体大小。

函数原型

std::vector<int64_t> GetDimSizes() const

参数说明

无

返回值说明

std::vector<int64_t>，Tensor的shape的值。

约束说明

无。

调用示例

std::vector<int64_t> dims = shape->GetDimSizes();