美文网首页
pytorch之torch.nn.lstm

pytorch之torch.nn.lstm

作者: 很菜_嗯 | 来源:发表于2019-03-28 09:59 被阅读0次

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

相关文章

网友评论

      本文标题:pytorch之torch.nn.lstm

      本文链接:https://www.haomeiwen.com/subject/ikyfbqtx.html