asc.language.core.get_shape_size

asc.language.core.get_shape_size(shape_info: ShapeInfo) → int

获取Shape中所有dim的累乘结果

对应的Ascend C函数原型

__aicore__ inline int GetShapeSize(const ShapeInfo& shapeInfo)

参数说明

• shape_info：ShapeInfo类型，LocalTensor或GlobalTensor的shape信息。

返回值说明

输入的shape_info中所有dim的累乘结果。

约束说明

无。

调用示例

size = asc.get_shape_size(shape_info)