Function _shuffle

imperative/python/megengine/random/rng.py:260–264

(inp: Tensor, seed: int, handle: int)

def _shuffle(inp: Tensor, seed: int, handle: int) -> Tensor:
    assert inp.size > 0, "size needs to be greater than 0"
    op = ShuffleRNG(seed=seed, handle=handle)
    output, _ = apply(op, inp)
    return output


def _exponential(