_VF.LSTM Implementation

yitianz · January 24, 2019, 8:52pm

Hello!

I’m trying to dig into the implementation of torch.nn.LSTM.

First I look at this file and see that there is a rnn_impls on line 197. Then I see it defined on lines 14-19. And then I go to _VF.py and see this.

Perhaps this is due to lack of understanding of types or VariableFunctions, but I’m confused as to where to go next to find where the actual functionality of LSTM is implemented.

kabbachokk · December 6, 2019, 5:06pm

github.com

pytorch/pytorch/blob/49777e67303f608987ec0948c7fd8f46f6d3ca83/torch/csrc/api/src/nn/modules/rnn.cpp

#include <torch/nn/modules/rnn.h>

#include <torch/nn/modules/dropout.h>
#include <torch/types.h>
#include <torch/utils.h>

#include <c10/util/Exception.h>

#include <array>
#include <cmath>
#include <cstdint>
#include <functional>
#include <memory>
#include <string>
#include <tuple>
#include <unordered_set>
#include <utility>
#include <vector>

namespace torch {

This file has been truncated. show original