nn.LSTMpython代码

以下是使用 PyTorch 中的 `nn.LSTM` 的示例代码：

import torch
import torch.nn as nn

# 定义输入数据
input_size = 10
sequence_length = 5
batch_size = 3

# 生成随机输入数据
input_data = torch.randn(sequence_length, batch_size, input_size)

# 定义 LSTM 模型
hidden_size = 20
num_layers = 2

lstm = nn.LSTM(input_size, hidden_size, num_layers)

# 初始化隐藏状态
h0 = torch.randn(num_layers, batch_size, hidden_size)
c0 = torch.randn(num_layers,＊＊＊