复现RNN的梯度消失 & 验证LSTM解决了梯度消失

背景

       RNNCell的计算公式  

       LSTMCell的计算公式

根据上面公式得出的梯度公式可得出RNN会导致梯度消失,但LSTM不会,本质的原因是RNN的参数只有W, LSTM的参数有W+b,b可以根据实际情况来调整,具体的求导公式什么的就不列出来了。

RNN梯度消失

import torch
import torch.nn as nn
import torch.optim as optim

# 定义一个简单的 RNNCell
class SimpleRNNCell(nn.Module):
    def __init__(self, input_size, hidden_size):
        super(SimpleRNNCe
评论
成就一亿技术人!
拼手气红包6.0元
还能输入1000个字符
 
红包 添加红包
表情包 插入表情
 条评论被折叠 查看
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值