torch.randint — PyTorch 1.10.1 documentation
pytorch.org › docs › stabletorch.randint. 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. With the global dtype default ( torch.float32 ), this function returns a tensor with dtype torch.int64. low ( int, optional) – Lowest integer to be ...