torch.nn.lstm()接受的数据输入是(序列长度,batchsize,输入维数),使用batch_first=True,可以使lstm接受维度为(batchsize,序列长度,输入维数)的数据输入,同时,lstm的输出数据维度也会变为batchsize放在第一维(可参考这篇博客)。

 

相关文章:

  • 2021-10-25
  • 2022-12-23
  • 2021-06-28
  • 2021-11-09
  • 2021-11-27
  • 2022-12-23
  • 2022-03-03
  • 2022-12-23
猜你喜欢
  • 2022-12-23
  • 2022-12-23
  • 2022-12-23
  • 2021-06-24
  • 2022-12-23
  • 2022-12-23
  • 2021-06-10
相关资源
相似解决方案