文档
注册

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();
搜索结果
找到“0”个结果

当前产品无相关内容

未找到相关内容,请尝试其他搜索词