check_shape_rule

对tensor的shape进行校验。

def check_shape_rule(shape, min_dim=None, max_dim=None, max_shape_num=None)

参数	说明
shape	需要校验的shape数据。需要为元素组或者列表，shape的每个值需为int类型且不能小于等于0。
min_dim	最小维数，默认为DEFAULT_MIN_SHAPE_DIM，即为1。
max_dim	最大维数，默认为DEFAULT_MAX_SHAPE_DIM，即为8。
max_shape_num	最大shape各维的值乘积。默认为DEFAULT_MAX_SHAPE_NUM，即为200000000。

无返回值。

如果校验失败，则抛RuntimeError。

shape数只能是大于等于min_dim，小于等于max_dim。

from tbe.common.utils import para_check 
para_check.check_shape_rule((32,64,64,3,54,32,21,12,33,32))

校验失败，会抛出RuntimeError。

父主题： 算子参数校验