check_shape_size

函数功能

检查shape乘积的大小是否大于或等于指定的限制值。

函数原型

def check_shape_size(shape, limit=SHAPE_SIZE_LIMIT+1)

参数说明

返回值说明

无

约束说明

如果校验的shape乘积大于等于指定的limit，则抛RuntimeError。

调用示例

from tbe.common.utils import para_check 
para_check.check_shape_size((32, 64, 64, 3))