diff options
Diffstat (limited to 'candle-nn/src/rnn.rs')
| -rw-r--r-- | candle-nn/src/rnn.rs | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/candle-nn/src/rnn.rs b/candle-nn/src/rnn.rs index 9f144cca..07795eda 100644 --- a/candle-nn/src/rnn.rs +++ b/candle-nn/src/rnn.rs @@ -197,7 +197,7 @@ impl RNN for LSTM { fn states_to_tensor(&self, states: &[Self::State]) -> Result<Tensor> { let states = states.iter().map(|s| s.h.clone()).collect::<Vec<_>>(); - Tensor::cat(&states, 1) + Tensor::stack(&states, 1) } } |