pytorch之torch.nn.lstm

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分类

©著作权归作者所有,转载或内容合作请联系作者
平台声明：文章内容（如有图片或视频亦包括在内）由作者上传并发布，文章内容仅代表作者本人观点，简书系信息发布平台，仅提供信息存储服务。

推荐阅读更多精彩内容