torch.nn 模块の循环层

在 pytorch 中,提供了三种循环层的实现

层对应的类功能
torch.nn.RNN()多层 RNN 单元
torch.nn.LSTM()多层长短期记忆 LSTM 单元
torch.nn.GRU()多层门限循环 GRU 单元
torch.nn.RNNCell()一个 RNN 循环层单元
torch.nn.LSTMCell()一个长短期记忆 LSTM 单元
torch.nn.GRUCell()一个门限循环 GRU 单元

以 torch.nn.RNN() 为例,介绍循环层的参数、输入和输出:
input_size: 输入 x 的特征数量
hidden_size: 隐层的特征数量
num_layers: RNN 网络的层数
nonlinearity:指定非线性函数使用 tanh 还是 relu,默认是 tanh
bias:如果是 False,那么 RNN 层就不会使用偏置权重,默认是 True
batch_first:如果是 True,那么输入和输出的 shape 应该是 [batch_size, time_step, feature]
dropout:如果值非零,那么除了最后一层外,其他 RNN 层的输出都会套上一个 dropout 层,默认为 0
bidirectional:如果是 True,将会编程一个双向 RNN,默认为 False

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值