【图像分类】基于PyTorch搭建GRU实现MNIST手写数字体识别(单/双向GRU,附完整代码和数据集)

写在前面:
首先感谢兄弟们的关注和订阅,让我有创作的动力,在创作过程我会尽最大能力,保证作品的质量,如果有问题,可以私信我,让我们携手共进,共创辉煌。

在https://blog.csdn.net/AugustMe/article/details/128969138文章中,我们使用了基于PyTorch搭建LSTM实现MNIST手写数字体识别,LSTM是单向的;
在https://blog.csdn.net/AugustMe/article/details/129011083文章中,我们使用了基于PyTorch搭建LSTM实现MNIST手写数字体识别,LSTM是双向的;
现在我们基于PyTorch搭建单向和双向GRU网络用于MNIST手写数字体识别,请注意查看GRU网络代码和LSTM网络代码的区别。

1.导入依赖库

导入torch用于构建GRU网络,导入matplotlib用于绘图,导入numpy用于计算:

import torch
from torch import nn, optim

你可能感兴趣的:(深度学习--图像分类,pytorch,分类,gru)