MDLSTM Implementation


I was trying to replicate this model, which uses MDLSTM cells (whose formulation I found here). However, I’m struggling to find any implementation and trying to bring this formulation into a custom layer of the PyTorch framework. I also found a thread in this forum about the topic, but didn’t get much from it.

So, I would like to ask if anyone knows if there is any implementation of the MDLSTM networks in this framework or knows how to implement their formulation.

Thanks in advance.