LSTM的参数解释
## LSTM的参数解释
- 共有7个参数,前三个是必须输入的
|参数名称|参数含义|参数示例|
|-|-|-|
|input_size|输入特征的维数|每一行输入元素的个数,[1,2,3]的input_size就是3|
|hidden_size|隐藏层状态的维数|隐藏层节点的个数|
|num_layers|LSTM堆叠的层数|默认为1|
|bias|隐藏层状态是否带有bias,默认为true|bias是偏置值|
|batch_first|输入输出的第一维是否为batch_size,默认值为False|如果LSTM的批量数据和dataloader中的批量数据是相同意义的,就设置为True,如果输入的数据是二维数据的话,也设置为True|
|dropout|是否在除最后一个RNN层外的其他RNN层后面加dropout层|默认值为0,输入值为0-1之间的小数,表示概率|
|bidirectional|是否是双向RNN,默认为false|若为 true,则:num_directions=2,否则为1|
- 输入参数格式
- input(seq_len,batch,input_size)
- h_0(num_layers*num_directions,batch,hidden_size)
- c_0(num_layers*num_directions,hidden_size)
- 输出参数格式
- output(seq_len,batch,num_directions*hidden_size)
- h_n(num_layers*num_directions,batch,hidden_size)
- c_n(num_layers*num_directions,batch,hidden_size)
