mirror of
https://github.com/huggingface/candle.git
synced 2025-06-15 18:28:24 +00:00
@ -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
|
||||
|
Reference in New Issue
Block a user