https://github.com/cnt-dev/pytorch-stateful-lstm
提供一个 Libtorch & C++ Extension 的 Demo。
总结：
- 用相同逻辑把 AllenNLP 的 LSTM Cell 重写了一下，性能提升 50%。
- 目前 Libtorch 的文档基本没有，而且官方文档与 1.0 的实现已经对不上了，建议想入坑的可以先等等。不过 Libtorch 的源码可读性挺好，没有文档问题不是太大的问题。
提供一个 Libtorch & C++ Extension 的 Demo。
总结：
- 用相同逻辑把 AllenNLP 的 LSTM Cell 重写了一下，性能提升 50%。
- 目前 Libtorch 的文档基本没有，而且官方文档与 1.0 的实现已经对不上了，建议想入坑的可以先等等。不过 Libtorch 的源码可读性挺好，没有文档问题不是太大的问题。
Select Language