对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。