写了一个pytorch框架下对LSTM的矩阵实现分块循环矩阵压缩的方法
参考这篇博客:https://blog.csdn.net/kuan__/article/details/116600433
2021-07-20 13:06:50
8KB
1
使用pytorch框架自定义了一个LSTM结构，压缩文件包含两个文件，一个是modules.py是编写的自定义LSTM结构，IMDB.py文件是使用modules.py里自定义的LSTM结构搭建的网络模型。pytorch自定义多层双向LSTM结构的程序详解可以参考这篇博客：https://blog.csdn.net/kuan__/article/details/114652053
2021-03-11 19:02:34
3KB
1