Struct std::RNNCell

#include <std/ml/rnn.h> source

struct RNNCell {
    &FloatTensor W_ih;
    &FloatTensor W_hh;
    &FloatTensor b_ih;
    &FloatTensor b_hh;
}

h_t = tanh(x_t @ W_ih + h_{t-1} @ W_hh + b_ih + b_hh)

Fields

§&FloatTensor W_ihsource

[inputSize, hiddenSize]

§&FloatTensor W_hhsource

[hiddenSize, hiddenSize]

§&FloatTensor b_ihsource

[1, hiddenSize]

§&FloatTensor b_hhsource

[1, hiddenSize]

§&FloatTensor forward(const &FloatTensor x_t, const &FloatTensor h_prev) constsource · definition

§void collect_params(&Vec<&FloatTensor> out) constsource · definition

Parameterized (module.h): every weight tensor.

§void collect_state(&Vec<&FloatTensor> out) constsource · definition

§void free() consumesource · definition

= drop()

§static struct RNNCell new(unsigned long inputSize, unsigned long hiddenSize)source · definition