Method get_recurrent_state

cake-core/src/models/common/cache.rs:221–223

Get the recurrent state for a linear attention layer.

(&self, block_idx: usize)

    /// Get the recurrent state for a linear attention layer.
    pub fn get_recurrent_state(&self, block_idx: usize) -> Option<&Tensor> {
        self.recurrent_states[block_idx].as_ref()
    }

    /// Set the recurrent state for a linear attention layer.
    pub fn set_recurrent_state(&mut self, block_idx: usize, state: Tensor) {