下载Pytorch自带的MNIST手写数字数据集

以下是如何下载Pytorch自带的MNIST手写数字数据集的示例代码：

import torch
import torchvision.datasets as datasets

# download the MNIST dataset
train_dataset = datasets.MNIST(root='./data', train=True, download=True)
test_dataset = datasets.MNIST(root='./data', train=False, download=True)

# print the number of samples in each dataset
print(f"Number of training samples: {len(train_dataset)}")
print(f"Number of test samples: {len(test_dataset)}")

在这个示例中，我们使用了`torchvision.datasets`模块中的`MNIST`类，它提供了访问MNIST数据集的方法。我们需要指定`root`参数来告诉它数据集应该被下载到哪个文件夹中。如果`train=True`，则下载训练数据集；如果`train=False`，则下载测试数据集。`download=True`告诉它从网络上下载数据集，如果数据集已经下载过，则不会重复下载。

一旦我们下载了数据集，我们可以使用`len`函数来打印数据集中样本的数量。在上面的代码中，我们分别打印了训练数据集和测试数据集的样本数量。