qq_53449784的博客

一、RNN实现结构原理代码实现import torchimport torch.nn as nnclass RNN(nn.Module): def __init__(self, input_size, hidden_size, output_size): super(RNN, self).__init__() self.hidden_size = hidden_size self.i2h = nn.Linear(inp.

一、准备数据代码import torchvisionimport torchvision.transforms as transformsfrom torch.utils.data import DataLoader# ========================================准备数据========================================# 定义预处理函数transform = transforms.Compose([transforms