torch.randint
torch.randint(low=0, high, size, *, generator=None, out=None, dtype=None, layout=torch.strided, device=None, requires_grad=False) -> Tensor
Returns a tensor filled with random integers generated uniformly
   between "low" (inclusive) and "high" (exclusive).
The shape of the tensor is defined by the variable argument "size".
Note:
 With the global dtype default ("torch.float32"), this function
 returns a tensor with dtype "torch.int64".

Parameters:
      * low (int, optional) -- Lowest integer to be drawn
        from the distribution. Default: 0.
  * **high** (*int*) -- One above the highest integer to be drawn
    from the distribution.

  * **size** (*tuple*) -- a tuple defining the shape of the output
    tensor.

Keyword Arguments:
      * generator ("torch.Generator", optional) -- a pseudorandom
        number generator for sampling
  * **out** (*Tensor**, **optional*) -- the output tensor.
