torch.randint_like
torch.randint_like(input, low=0, high, *, dtype=None, layout=torch.strided, device=None, requires_grad=False, memory_format=torch.preserve_format) -> Tensor
Returns a tensor with the same shape as Tensor "input" filled with
   random integers generated uniformly between "low" (inclusive) and
   "high" (exclusive).
Parameters:
      * input (Tensor) -- the size of "input" will determine
        size of the output tensor.
  * **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.

Keyword Arguments:
      * dtype ("torch.dtype", optional) -- the desired data type
        of returned Tensor. Default: if "None", defaults to the dtype
        of "input".
  * **layout** ("torch.layout", optional) -- the desired layout of
    returned tensor. Default: if "None", defaults to the layout of
