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。
父主题： 算子参数校验