huggingface/candle
1
0
Fork 0
mirror of https://github.com/huggingface/candle.git synced 2025-06-16 10:38:54 +00:00
Code Issues Packages Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
Justin Sing
2024-08-01 10:30:32 -04:00
committed by GitHub
parent 9ca277a9d7
commit 6991a37b94
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
candle-nn/src/rnn.rs
View File

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