def init(shape):
    rng = np.random.RandomState(seed)
    return create_cumprod_input(rng, shape, axis, with_mask)
def init(shape):
    rng = np.random.RandomState(seed)
    return create_cumprod_input(rng, shape, axis, with_mask)