2021
07-26
07-26
pytorch lstm gru rnn 得到每个state输出的操作
默认只返回最后一个state,所以一次输入一个step的input#coding=UTF-8importtorchimporttorch.autogradasautograd#torch中自动计算梯度模块importtorch.nnasnn#神经网络模块torch.manual_seed(1)#lstm单元输入和输出维度都是3lstm=nn.LSTM(input_size=3,hidden_size=3)#生成一个长度为5,每一个元素为1*3的序列作为输入,这里的数字3对应于上句中第一个3inputs=[autograd.Variable(torch.randn((1,3)))...
继续阅读 >