检查shape乘积的大小是否超过指定限制limit。
def check_shape_size(shape, limit=SHAPE_SIZE_LIMIT+1)
参数 |
说明 |
|---|---|
shape |
需要校验的shape数据。 |
limit |
指定的限制值，默认为SHAPE_SIZE_LIMIT+1。其中SHAPE_SIZE_LIMIT为= 2 ** 31 - 1。 |
无
如果校验的shape乘积超过指定的limit，则抛RuntimeError。
from te.utils import para_check para_check.check_shape_size((32, 64, 64, 3))
结果：校验通过。