19 #ifndef TESSERACT_LSTM_LSTM_H_ 20 #define TESSERACT_LSTM_LSTM_H_ 50 LSTM(
const STRING&
name,
int num_inputs,
int num_states,
int num_outputs,
68 if (softmax_ != NULL) spec += softmax_->
spec();
105 virtual void Update(
float learning_rate,
float momentum,
int num_samples);
110 double* changed)
const;
123 void ResizeForward(
const NetworkIO& input);
160 #endif // TESSERACT_LSTM_LSTM_H_ virtual void CountAlternators(const Network &other, double *same, double *changed) const
void add_str_int(const char *str, int number)
virtual STRING spec() const
virtual STRING spec() const
virtual void ConvertToInt()
virtual bool Backward(bool debug, const NetworkIO &fwd_deltas, NetworkScratch *scratch, NetworkIO *back_deltas)
virtual void Update(float learning_rate, float momentum, int num_samples)
virtual bool DeSerialize(TFile *fp)
LSTM(const STRING &name, int num_inputs, int num_states, int num_outputs, bool two_dimensional, NetworkType type)
virtual int InitWeights(float range, TRand *randomizer)
virtual void SetEnableTraining(TrainingState state)
virtual void DebugWeights()
const STRING & name() const
virtual void Forward(bool debug, const NetworkIO &input, const TransposedArray *input_transpose, NetworkScratch *scratch, NetworkIO *output)
virtual bool Serialize(TFile *fp) const
virtual StaticShape OutputShape(const StaticShape &input_shape) const