https://morvanzhou.github.io/tutorials/machine-learning/torch/3-05-train-on-batch/
DataLoader 以及BATCH_SIZE问题
torch.nn.LSTM()输入API.
重要参数:
input_size:每个时步(time-step)输入到lstm单元的维度。(实际输入的数据size为[batch_size, input_size])
hidden_size: 确定了隐含状态hidden_state的维度.
num_layers: 叠加的层数。
示例代码:
使用单向LSTM进行MNIST分类
![](https://img.haomeiwen.com/i14460008/597756a0172e3b00.png)
![](https://img.haomeiwen.com/i14460008/f3c61bf7e4fe16ed.png)
![](https://img.haomeiwen.com/i14460008/caa1ceb852855f71.png)
![](https://img.haomeiwen.com/i14460008/03a1307bfa719cc0.png)
网友评论