torch.nn.LSTMCell详解代码

最新推荐文章于 2024-01-05 10:24:16 发布

求学路上的小白

最新推荐文章于 2024-01-05 10:24:16 发布

阅读量1.3k

点赞数

分类专栏： pytorch 文章标签： pytorch

本文链接：https://blog.csdn.net/qq_37150124/article/details/109611786

版权

pytorch 专栏收录该内容

8 篇文章 1 订阅

订阅专栏

小白撸代码

import torch.nn as nn
import torch
from torch.autograd import Variable
#LSTMCell输入10维度 隐藏状态20维度 
#规定了下面输入的列数和隐藏层的列数
rnn = nn.LSTMCell(10,20)
#随机输入6行大矩阵 每行大矩阵为3行10列；10为LSMTMCell的10
input = Variable(torch.randn(6,3,10)) 
# print(input)
#保存着batch中每个元素的初始化隐状态的Tensor
#隐藏层为3行20列矩阵；20为LSTMCell中的20
hx = Variable(torch.randn(3,20))
#保存着batch中每个元素的初始化细胞状态的Tensor
#细胞状态为3行20列矩阵；20 为LSTMCell中的20
cx = Variable(torch.randn(3,20))
#输出
output = []
for i in range(6):#将输出格式变为（6，3，20）
    #下一个时刻的隐状态。下一时刻细胞状态
    hx ,cx = rnn(input[i],(hx,cx))
    output.append(hx)
print(output)