Random Int Tensor Pytorch