笔记53:torch.nn.rnn() 函数详解

笔记53:torch.nn.rnn() 函数详解_第1张图片

参数解释:

(1)input_size():即输入信息 Xt 的每个序列的独热编码向量的长度,即 len(vocab)

(2)hidden_size():即隐变量 h 的维度(维度是多少,就代表用几个数字存储该序列的历史信息)

(3)num_layers:隐藏层的层数

a

a

a

在视频课中:

batch_size = 32

input_size = len(vocab) = 28

hidden_size = 256

num_layers = 1

笔记53:torch.nn.rnn() 函数详解_第2张图片

你可能感兴趣的:(笔记,rnn,深度学习)