Method sample_one

test/backend/test_randomness.py:370–370

()

    @TinyJit
    def sample_one(): return Tensor(w).multinomial(1, replacement=False).realize()

    tiny_samples = [sample_one().item() for _ in range(400)]
    torch_samples = [torch.tensor(w).multinomial(1, replacement=False).item() for _ in range(400)]