Pytorch十二讲 RNN基础（部分笔记）

第12讲 循环神经网络基础篇（部分笔记）

B站 刘二大人 传送门：《PyTorch深度学习实践》完结合集

仅把RNN代码复制了过来

################################################################################
# 刘二大人 RNN基础test1
# 1.RNNCELL
################################################################################
import torch
batch_size = 1
seq_len = 3
input_size = 4
hidden_size = 2
cell = torch.nn.RNNCell(input_size=input_size, hidden_size=hidden_size)

dataset = torch.randn(seq_len, batch_size, input_size)
hidden = torch.zeros(batch_size, hidden_size)
for idx, input in enumerate(dataset):
    print('='*20, idx, '='*20)
    print('Input size:', input.shape)
    hidden = cell(input, hidden)          # 当前时刻的输入及hidden
    # input of shape (batch_size, input_size)
    #