Torch.nn.functional.lstm


(Armin Li) #1

Hi, all

I want to use a functional lstm, just as torch.nn.functional.linear, how to reach that?

Thanks.


#2

Just implement the function yourself. It will take you a few mins. You can find the equations of the LSTM
in the docs under nn.LSTM .