sample_idx = torch.randint(len(training_data), size=(1,)).item() img, label = training_data[sample_idx] figure.add_subplot(rows, cols, i) plt.title(labels_map[label]) plt.axis("off") plt.imshow(img.squeeze(), cmap="gray")

这段代码的作用是从 `training_data` 数据集中随机选择一个样本，并将其对应的图像和标签显示在 Matplotlib 中的一个子图中。

具体来说，前一行代码的作用已经解释过了，即随机选择一个样本并返回其在数据集中的索引。接下来的一行代码中，将随机选择的样本通过 `training_data[sample_idx]` 语句获取，其中 `img` 表示该样本的图像，`label` 表示该样本的标签。接下来，将该子图添加到 Matplotlib 的 `figure` 对象中，并设置其在子图矩阵中的位置。然后，设置该子图的标题为该样本的标签，将坐标轴关闭，并使用 `imshow()` 方法将该样本的图像显示出来。最后，将 `img` 转换为灰度图像，并使用 `cmap` 参数设置颜色映射。

这段代码通常用于可视化数据集中的样本，以便更好地理解数据集的特点和结构。

相关问题

sample_idx = torch.randint(len(training_data), size=(1,)).item()

这段代码的作用是从 `training_data` 数据集中随机选择一个样本，并返回该样本在数据集中的索引。具体来说，它使用 PyTorch 中的 `torch.randint()` 函数生成一个随机整数，该随机整数的范围是 0 到 `training_data` 数据集的长度减1，然后使用 `.item()` 方法将该随机整数转换为 Python 中的标量值。这个随机整数就是随机选择的样本在数据集中的索引。

rand_idx = random.randint（0，20）

这段代码的作用是随机生成一个介于 0 和 20 之间的整数，并将其赋值给变量 `rand_idx`。具体来说，`random.randint(0, 20)` 会生成一个随机整数，这个整数的范围是从 0 到 20（包括 0 和 20）之间的整数。因此，`rand_idx` 的值会是一个介于 0 和 20 之间的整数。