2 个不稳定版本
新版本 0.2.0 | 2024 年 8 月 25 日 |
---|---|
0.1.0 | 2024 年 8 月 24 日 |
#544 在 机器学习 中
每月下载量 250
23KB
511 行
Candle LSTM
重新实现 Candle LSTM 推理以提高推理速度,包括双向 LSTM。
测试数据
安装 Pytorch 并运行 simple.py 生成测试数据。
- lstm_test.pt: Pytorch LSTM,batch_first = False。
- lstm_test_batch_first.pt: Pytorch LSTM,batch_first = True。
- bi_lstm_test.pt: Pytorch 双向 LSTM,batch_first = False。
- bi_lstm_test_batch_first.pt: Pytorch 双向 LSTM,batch_first = True。
依赖关系
~11MB
~218K SLoC