Template Class RNNCellImplBase

• Defined in File rnn.h

Inheritance Relationships

Base Type

• public torch::nn::Cloneable< Derived > (Template Class Cloneable)

Class Documentation

template<typename Derived>
class RNNCellImplBase : public torch::nn::Cloneable<Derived>

Base class for all RNNCell implementations (intended for code sharing).

Public Functions

explicit RNNCellImplBase(const RNNCellOptionsBase &options_)

virtual void reset() override

Initializes the parameters of the RNNCell module.

void reset_parameters()

virtual void pretty_print(std::ostream &stream) const override

Pretty prints the RNN module into the given stream.

Public Members

RNNCellOptionsBase options_base

Tensor weight_ih

Tensor weight_hh

Tensor bias_ih

Tensor bias_hh

Protected Functions

void check_forward_input(const Tensor &input, const std::string name) const

virtual std::string get_nonlinearity_str() const