Shortcuts

Template Class RNNCellImplBase

Inheritance Relationships

Base Type

Class Documentation

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

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

Public Functions

RNNCellImplBase(const RNNCellOptionsBase &options_)
void reset() override

Initializes the parameters of the RNNCell module.

void reset_parameters()
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
void check_forward_hidden(const Tensor &input, const Tensor &hx, std::string hidden_label) const
std::string get_nonlinearity_str() const

Docs

Access comprehensive developer documentation for PyTorch

View Docs

Tutorials

Get in-depth tutorials for beginners and advanced developers

View Tutorials

Resources

Find development resources and get your questions answered

View Resources