pytorch LSTM输入参数

PyTorch中的LSTM的输入参数包括：

1. input_size：每个时间步的输入特征维度大小。

2. hidden_size：LSTM单元中隐藏状态的维度大小。

3. num_layers：LSTM网络中的层数。

4. bias：是否添加偏置。

5. batch_first：如果设置为True，输入和输出张量中的第一维将是批次大小。

6. dropout：在LSTM单元内部使用的丢弃率，防止过拟合。

7. bidirectional：是否使用双向LSTM。

8. input：输入序列，形状为(seq_len, batch, input_size)或(batch, seq_len, input_size)。

9. h_0：LSTM的初始隐藏状态，形状为(num_layers * num_directions, batch, hidden_size)。

10. c_0：LSTM的初始单元状态，形状为(num_layers * num_directions, batch, hidden_size)。

注意：其中的num_directions等于1或2，表示单向或双向LSTM。