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)#