如何在Pytorch中实现具有多个单元的LSTM层？

1 回答

• 0

  这些单元格将根据输入中的序列大小自动展开 . 请查看此代码：

  # One cell RNN input_dim (4) -> output_dim (2). sequence: 5, batch 3
  # 3 batches 'hello', 'eolll', 'lleel'
  # rank = (3, 5, 4)
  inputs = Variable(torch.Tensor([[h, e, l, l, o],
                                  [e, o, l, l, l],
                                  [l, l, e, e, l]]))
  print("input size", inputs.size())  # input size torch.Size([3, 5, 4])
  
  # Propagate input through RNN
  # Input: (batch, seq_len, input_size) when batch_first=True
  # B x S x I
  out, hidden = cell(inputs, hidden)
  print("out size", out.size())  # out size torch.Size([3, 5, 2])
  

  您可以在https://github.com/hunkim/PyTorchZeroToAll/找到更多示例 .

  回复于 2024-04-28T14:08:14+08:00