随机采样函数torch.randint()

def randn_sampling(maxint, sample_size, batch_size):
    return torch.randint(maxint, size=(batch_size, sample_size, 2))

使用 PyTorch 库生成一个包含随机整数的张量。函数接受三个参数：

1. maxint：生成的随机整数的最大值（不包括该值）。
2. sample_size：每个批次中每个样本的随机整数数量。
3. batch_size：生成的批次数量。

 在函数内部，使用 torch.randint 函数生成一个张量，该张量的形状为 (batch_size, sample_size, 2)，其中每个元素是一个随机整数。