深度学习基础题1：编写一个简单的图像分类器

深度学习基础题1：编写一个简单的图像分类器

深度学习已经成为计算机视觉和人工智能领域中的重要技术，图像分类是其中一个常见的任务。在这道题目中，我们将编写一个简单的图像分类器，使用深度学习模型对图像进行分类。

首先，我们需要准备数据集。本例中，我们将使用一个公共的数据集——MNIST手写数字数据集。该数据集包含了一系列28x28像素的灰度图像，每个图像都代表一个手写数字（0到9之间）。我们需要使用这个数据集进行模型的训练和测试。

接下来，我们将使用Python编写代码来实现图像分类器。首先，我们需要导入所需的库。

import torch
import torchvision
import torchvision.transforms as transforms
import torch.nn as nn