背景

根据上面公式得出的梯度公式可得出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


最低0.47元/天 解锁文章
1229






