Struct LSTMCellOptions¶
Defined in File rnn.h
Page Contents
Struct Documentation¶
-
struct LSTMCellOptions¶
Options for the
LSTMCell
module.Example:
LSTMCell model(LSTMCellOptions(20, 10).bias(false));
Public Functions
-
LSTMCellOptions(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¶
The number of features in the hidden state
h
-
inline auto bias(const bool &new_bias) -> decltype(*this)¶
If
false
, then the layer does not use bias weightsb_ih
andb_hh
.Default:
true
-
inline auto bias(bool &&new_bias) -> decltype(*this)¶
-
inline const bool &bias() const noexcept¶
-
inline bool &bias() noexcept¶
-
LSTMCellOptions(int64_t input_size, int64_t hidden_size)¶