diff options
Diffstat (limited to 'candle-nn')
| -rw-r--r-- | candle-nn/src/rnn.rs | 4 |
1 files changed, 4 insertions, 0 deletions
diff --git a/candle-nn/src/rnn.rs b/candle-nn/src/rnn.rs index 30ad6ff5..5934af85 100644 --- a/candle-nn/src/rnn.rs +++ b/candle-nn/src/rnn.rs @@ -55,6 +55,10 @@ pub struct LSTMState { } impl LSTMState { + pub fn new(h: Tensor, c: Tensor) -> Self { + LSTMState { h, c } + } + /// The hidden state vector, which is also the output of the LSTM. pub fn h(&self) -> &Tensor { &self.h |