3rdparty/caffe/docs/tutorial/layers/lstm.md

metadata

title: LSTM Layer

LSTM Layer

• Layer type: LSTM
• Doxygen Documentation
• Header: ./include/caffe/layers/lstm_layer.hpp
• CPU implementation: ./src/caffe/layers/lstm_layer.cpp
• CPU implementation (helper): ./src/caffe/layers/lstm_unit_layer.cpp
• CUDA GPU implementation (helper): ./src/caffe/layers/lstm_unit_layer.cu

Parameters

• Parameters (Parameter recurrent_param)
• From ./src/caffe/proto/caffe.proto:

{% highlight Protobuf %} {% include proto/RecurrentParameter.txt %} {% endhighlight %}