PyTorch-LSTM

import torch
import torch.nn as nn

torch.random.manual_seed(10)

input_size = 2  # 输入向量维度
hidden_size = 4 # 隐层层维度
num_layers = 2 # 层数

lstm = nn.LSTM(input_size, hidden_size, num_layers)


# Input:

# input of shape (sep_len, bath, input_size)
# h_t-1 of shape (num_directions * num_layers, bath, hidden_size)
# c_t-1 for shape (num_directions * num_layers, bath, hidden_size)

# Output:
# output of shape (sep_len, bath, num_directions * hidden_size)
# h_t-1 of shape (num_directions * num_layers, bath, hidden_size)
# c_t-1 for shape (num_directions * num_layers, bath, hidden_size)

# two ways
Input = torch.randn(4, 3, 2)
h = torch.randn(2, 3, 4)
c = torch.randn(2, 3, 4)
output = None

# first
h1 = h
c1 = c
for it in Input:
    output, (h1, c1) = lstm(it.view(1, 3, -1), (h1, c1))
    print((output == h1[-1]).all().item())
print(output)

# second
output1, (h, c) = lstm(Input,(h, c))
print(output1[-1])
# print(output1[-1] == output) 精度的问题