目录

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