update: LSTMState and GRUState fields to be public (#2384)

2024-08-01 10:30:32 -04:00 · 2024-08-01 10:30:32 -04:00 · 6991a37b94
parent 9ca277a9d7
commit 6991a37b94
1 changed files with 3 additions and 3 deletions
--- a/candle-nn/src/rnn.rs
+++ b/candle-nn/src/rnn.rs
@ -50,8 +50,8 @@ pub trait RNN {
 #[allow(clippy::upper_case_acronyms)]
 #[derive(Debug, Clone)]
 pub struct LSTMState {
-    h: Tensor,
-    c: Tensor,
+    pub h: Tensor,
+    pub c: Tensor,
 }

 impl LSTMState {
@ -205,7 +205,7 @@ impl RNN for LSTM {
 #[allow(clippy::upper_case_acronyms)]
 #[derive(Debug, Clone)]
 pub struct GRUState {
-    h: Tensor,
+    pub h: Tensor,
 }

 impl GRUState {