gru pytorch案例

下面是一个使用PyTorch实现GRU的例子：

首先，我们需要导入相关的PyTorch模块：

import torch
import torch.nn as nn

然后，定义一个GRU模型：

class GRUModel(nn.Module):
    def __init__(self, input_size, hidden_size, output_size):
        super(GRUModel, self).__init__()
        self.hidden_size = hidden_size
        self.gru = nn.GRU(input_size, hidden_size)
        self.fc = nn.Linear(hidden_size, output_size)
    
    def forward(self, input, hidden):
        output, hidden = self.gru(input, hidden)
        output = self.fc(output)
        return output, hidden
    
    def init_hidden(self, batch_size):
        return torch.zeros(1, batch_size, self.hidden_size)

在这个模型中，我们定义了一个GRU层和一个全连接层。在forward函数中，我们首先将输入和隐藏状态传递给GRU层，然后将输出传递给全连接层。在init_hidden函数中，我们初始化了一个大小为(batch_size, hidden_size)的隐藏状态张量。

接下来，我们需要定义一个损失函数和优化器：

criterion = nn.CrossEntropyLoss()
optimizer = torch.optim.Adam(model.parameters(), lr=learning_rate)

然后，我们可以开始训练模型了：

for epoch in range(num_epochs):
    losses = []
    hidden = model.init_hidden(batch_size)
    for i in range(0, input_size - sequence_length, sequence_length):
        inputs = torch.Tensor([one_hot_encoded[j:j+sequence_length] for j in range(i,i+sequence_length)])
        targets = torch.LongTensor([encoded[j+sequence_length] for j in range(i,i+sequence_length)])
        
        optimizer.zero_grad()
        
        output, hidden = model(inputs.view(sequence_length, batch_size, -1), hidden)
        loss = criterion(output.view(sequence_length, -1), targets)
        loss.backward()
        optimizer.step()
        
        losses.append(loss.item())
        
    if epoch % 10 == 0:
        print(f'Epoch {epoch}, Loss: {sum(losses)/len(losses):.4f}')

在这个训练循环中，我们首先初始化了一个隐藏状态。然后，我们将数据分成一系列序列，并将它们转换为one-hot编码的张量。接下来，我们将输入和隐藏状态传递给模型，并计算损失。最后，我们使用反向传播来更新模型的参数。

最后，我们可以使用训练好的模型进行预测：

def predict(model, input_str, char2int, int2char):
    input = torch.Tensor([one_hot_encode(input_str, char2int)])
    hidden = model.init_hidden(1)
    output, hidden = model(input.view(len(input_str), 1, -1), hidden)
    output = output.view(-1, output_size)
    output = torch.argmax(output, dim=1)
    output_str = ''.join([int2char[i] for i in output])
    return output_str

在这个函数中，我们首先将输入字符串转换为one-hot编码的张量，然后将其传递给模型进行预测。最后，我们将预测结果转换回字符形式。

这就是一个使用PyTorch实现GRU的例子。