Pytorch LSTM2

 

import torch
from torch.autograd import Variable
from torch import nn

mylstm = nn.LSTM(input_size=100,hidden_size=100,num_layers=1,bidirectional=False)
for p in mylstm.parameters():
    print(p.size())

 
lstm_seq = nn.LSTM(28, 4, num_layers=2)  # 构建LSTM网络
for p in lstm_seq.parameters():
    print(p.size())

lstm_input = Variable(torch.randn(10, 3, 28))  # 构建输入
h_init = Variable(torch.randn(2, 3, 4))        # 构建h输入参数
c_init = Variable(torch.randn(2, 3, 4))        # 构建c输出参数
out, (h, c) = lstm_seq(lstm_input, (h_init, c_init))    # 计算

print(lstm_seq.weight_ih_l0.shape)
print(lstm_seq.weight_hh_l0.shape)
print(out.shape, h.shape, c.shape)

 

 

Reference

1 missing