| 9 | } |
| 10 | |
| 11 | torch::Tensor LSTM::forward(torch::Tensor x){ |
| 12 | auto lstm_out = lstm->forward(x); |
| 13 | auto predictions = ln->forward(std::get<0>(lstm_out)); |
| 14 | return predictions.select(1,-1); |
| 15 | } |
nothing calls this directly
no outgoing calls
no test coverage detected