Struct Cache

pub struct Cache { /* private fields */ }

Implementations

impl Cache

pub fn new(dim: usize, max_seq_len: usize) -> Self

pub fn dim(&self) -> usize

pub fn current_seq_len(&self) -> usize

pub fn max_seq_len(&self) -> usize

pub fn all_data(&self) -> &Option<Tensor>

pub fn current_data(&self) -> Result<Option<Tensor>>

pub fn reset(&mut self)

pub fn append(&mut self, src: &Tensor) -> Result<()>

Trait Implementations

impl Clone for Cache

fn clone(&self) -> Cache

Returns a copy of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl Debug for Cache

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.