Thank you for your answer, but that is not exactly what I’m asking.
What I’m asking is:
In this example - https://pytorch.org/tutorials/advanced/cpp_extension.html, we see an example of LLTM class. In this example “forward” and “backward” methods are implemented in cpp and called from python “forward” and “backward” methods, instead of the python implementation.
Is it possible to write the whole LLTM class in cpp and use it from python, instead of just the methods?