下载
EN
注册
我要评分
文档获取效率
文档正确性
内容完整性
文档易理解
在线提单
论坛求助
昇腾小AI

GetShapeSize

函数功能

获取当前Tensor运行时的shape大小,即此Tensor中包含的元素的数量。

函数原型

int64_t GetShapeSize() const

参数说明

不涉及

返回值说明

返回执行时shape的大小。

约束说明

无。

调用示例

Tensor tensor{{{8, 3, 224, 224}, {16, 3, 224, 224}},       // shape                
              {ge::FORMAT_ND, ge::FORMAT_FRACTAL_NZ, {}},  // format                
              kOnDeviceHbm,                                // placement                
              ge::DT_FLOAT16,                              //dt                
              nullptr};
auto shape_size = tensor.GetShapeSize(); // 16*3*224*224
搜索结果
找到“0”个结果

当前产品无相关内容

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