pytorch学习笔记（七）

最新推荐文章于 2024-08-19 20:03:28 发布

迷茫，我太迷茫了！

最新推荐文章于 2024-08-19 20:03:28 发布

阅读量197

点赞数

分类专栏：研究生入门阶段文章标签： pytorch 学习深度学习

本文链接：https://blog.csdn.net/qq_41730733/article/details/125867827

版权

本文是PyTorch学习笔记的第七篇，主要探讨RNNcell和RNN的区别以及相关参数。内容包括RNNcell的工作原理，RNN结构的注意点，以及如何将输入x和隐藏状态h进行线性变换。

摘要由CSDN通过智能技术生成

RNN的恩怨情仇

【注】x的维度是input_size,对x做线性变换的权重Wih为（hidden_size,input_size）

hidden的size为hidden_size,对hidden做线性变换的权重Whh为（hidden_size,hidden_size。

他俩能拼接起来【 w1,w2】[h与x列拼接]，输出结果的size为（hidden_size,1）

RNNcell()

batch_size = 1
seq_len = 3
input_size = 4
hidden_size = 2
cell = torch.nn.RNNCell(input_size= input_size,hidden_size=hidden_size)
hidden = torch.zeros(batch_size,hidden_size)
dataset = torch.randn(seq_len,batch_size,input_size)
print('dataset',dataset)
for idx,input in enumerate(dataset):
    print('='*20,idx)
    print("input",input)
    print("input_size",input_size)
    hidden = cell(input,hidden)
    print('hiden_size',hidden_size)
    print('hidden'