nn.LSTM 参数

nn.LSTM 是 PyTorch 中的一个模块，用于实现长短期记忆（LSTM）网络。它有以下参数：

- input_size：输入数据的特征维度。
- hidden_size：隐藏状态的特征维度。
- num_layers：LSTM 层数。
- bias：是否使用偏置。
- batch_first：如果设置为 True，则输入和输出张量的第一个维度是批次大小，而不是时间步长。
- dropout：如果非零，则在每个 LSTM 层的输出上应用 dropout 操作，以防止过拟合。
- bidirectional：如果设置为 True，则使用双向 LSTM。

以下是一个示例：

import torch.nn as nn

input_size = 10
hidden_size = 20
num_layers = 2
batch_first = True
bidirectional = True

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

在这个示例中，我们创建了一个具有输入特征维度为 10，隐藏状态特征维度为 20，双向 LSTM 的模型，该模型有两个 LSTM 层，并且输入和输出张量的第一个维度是批次大小。