pytorch torch.randint

这段代码使用 PyTorch 创建一个包含随机整数的张量。具体来说：

label = torch.randint(10, (bs,))

以下是这段代码的详细解释：

1. torch.randint:

   • 这是 PyTorch 中用于生成随机整数的函数。
   • torch.randint(low, high, size) 会生成一个包含在 [low, high) 区间内的随机整数的张量。

2. 参数解释:

   • 10：表示生成的随机整数在 [0, 10) 区间内（包括 0，但不包括 10）。
   • (bs,)：表示生成的张量的形状。这里 (bs,) 是一个一维张量，长度为 bs。

3. label:

   • 生成的张量被赋值给变量 label。

假设 bs 是批量大小（batch size），例如 bs = 4，那么这段代码会生成一个形状为 [4] 的张量，其中包含 4 个在 [0, 10) 区间内的随机整数。

例如，如果 bs = 4，可能的输出是：

tensor([3, 7, 1, 9])