在Python中实现LSTM单元的修改架构

Question

我希望修改标准LSTM单元表示的内部单元结构和方程（例如在Keras的实现中）。我该怎么做（要覆盖哪些功能/模块以及Keras实现的哪个文件）？ 关于图书馆或框架的任何其他建议也是受欢迎的。

Answer 1

您想要创建一个新的LSTM单元类，您可以在keras github repo上找到源代码。之后，您需要创建一个用作新图层的新类，并从此自定义图层中的 init 图层调用您的单元格。目前还不清楚你想要做什么，但我可以推荐阅读：

This this tutorial和this code。