def _check_shape_contain_zero(shp):
    """Check whether shape contains 0"""
    if isinstance(shp, int):
        return shp == 0
    if isinstance(shp, (list, tuple)):
        for s in shp:
            if s == 0:
                return True
    return False