def create_cumprod_input(rng, shape, axis, with_mask):
    x = rng.randn(*shape).astype(np.float32)
    if with_mask:
        mask = (rng.rand(*shape) > (1.0 / shape[axis]))
        x = (x * mask)
    return x
def create_cumprod_input(rng, shape, axis, with_mask):
    x = rng.randn(*shape).astype(np.float32)
    if with_mask:
        mask = (rng.rand(*shape) > (1.0 / shape[axis]))
        x = (x * mask)
    return x