def wrapped(*args_, **kwargs_):
    if np.random.binomial(1, p):
        return func(*args_, *args, **kwargs_, **kwargs)
    return squeeze_first(args_)
def wrapped(*args_, **kwargs_):
    if np.random.binomial(1, p):
        return func(*args_, *args, **kwargs_, **kwargs)
    return squeeze_first(args_)