第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)
#