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。
父主题: shape相关工具