check_shape

paddle. check_shape ( shape: ShapeLike, op_name: str, expected_shape_type: _ClassInfo = (<class 'list'>, <class 'tuple'>, <class 'paddle.base.framework.Variable'>, <class 'paddle.base.libpaddle.pir.Value'>), expected_element_type: _ClassInfo = (<class 'int'>, <class 'paddle.base.framework.Variable'>, <class 'paddle.base.libpaddle.pir.Value'>), expected_tensor_dtype: tuple[_DTypeLiteral, ...] = ('int32', 'int64') ) None [source]