下载
中文
注册

get_shape_size

函数功能

获取shape的各维度值的乘积。

函数原型

def get_shape_size(shape)

参数说明

参数

说明

shape

数据shape

返回值说明

返回shape各维度值的乘积。

约束说明

调用示例

from tbe.common.utils import shape_util 
shape_util.get_shape_size((1,2,3,4)) 

获取shape(1,2,3,4)的各维度乘积,返回24。