用pytorch写一段rnn代码用来进行电力负荷预测，并提供数据集

在PyTorch中，我们可以使用循环神经网络（RNN）进行电力负荷预测，这是一种适合处理序列数据的时间序列预测任务。这里是一个简单的例子，展示如何创建一个基本的LSTM模型：

import torch
import torch.nn as nn

# 定义一个LSTM模型
class PowerLoadPredictor(nn.Module):
    def __init__(self, input_size, hidden_size, num_layers, output_size):
        super(PowerLoadPredictor, self).__init__()
        self.hidden_size = hidden_size
        self.num_layers = num_layers
        self.lstm = nn.LSTM(input_size, hidden_size, num_layers, batch_first=True)
        self.fc = nn.Linear(hidden_size, output_size)

    def forward(self, x, h0=None):
        # x: (batch_size, seq_len, input_size)
        out, _ = self.lstm(x, h0)
        # out: (batch_size, seq_len, hidden_size)
        out = self.fc(out[:, -1, :])  # 取出最后一个时间步的隐藏状态作为输入到全连接层
        return out

# 示例参数
input_size = 24  # 输入特征的数量（一天24小时）
hidden_size = 64
num_layers = 2
output_size = 1  # 预测值通常是一个数值，这里是电力负荷的估计

# 初始化模型
model = PowerLoadPredictor(input_size, hidden_size, num_layers, output_size)

# 假设我们有一个电力负荷的数据集（例如用电量每小时一列）
# 实际上你需要准备一个Tensor，形状应该是(batch_size, sequence_length, input_size)，
# 例如: data_loader是一个PyTorch的DataLoader实例，加载了训练数据
data_loader = ...  # 负荷数据的加载函数，需要填充实际数据集

for inputs in data_loader:
    # 假设inputs是一个包含时间序列数据的张量
    outputs = model(inputs)