Shortcuts

Struct GRUCellOptions

Page Contents

Struct Documentation

struct GRUCellOptions

Options for the GRUCell module.

Example:

GRUCell model(GRUCellOptions(20, 10).bias(false));

Public Functions

GRUCellOptions(int64_t input_size, int64_t hidden_size)
inline auto input_size(const int64_t &new_input_size) -> decltype(*this)

The number of expected features in the input x

inline auto input_size(int64_t &&new_input_size) -> decltype(*this)
inline const int64_t &input_size() const noexcept
inline int64_t &input_size() noexcept
inline auto hidden_size(const int64_t &new_hidden_size) -> decltype(*this)

The number of features in the hidden state h

inline auto hidden_size(int64_t &&new_hidden_size) -> decltype(*this)
inline const int64_t &hidden_size() const noexcept
inline int64_t &hidden_size() noexcept
inline auto bias(const bool &new_bias) -> decltype(*this)

If false, then the layer does not use bias weights b_ih and b_hh.

Default: true

inline auto bias(bool &&new_bias) -> decltype(*this)
inline const bool &bias() const noexcept
inline bool &bias() noexcept

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