Method get_conv_state

cake-core/src/models/common/cache.rs:231–233

Get the conv state for a linear attention layer.

(&self, block_idx: usize)

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

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