Struct TensorPersistentCache 

pub struct TensorPersistentCache { /* private fields */ }

Persistent cache that works with byte arrays for tensor data

Implementations

impl TensorPersistentCache

pub fn new<P: AsRef<Path>>(cache_dir: P, capacity: usize) -> Result<Self>

Create a new tensor persistent cache

pub fn get<T>( &mut self, index: &usize, ) -> Result<Option<(Tensor<T>, Tensor<T>)>>

where T: Clone + Default + Zero + Send + Sync + 'static + NumCast,

Get tensors from cache

pub fn insert<T>( &mut self, index: usize, features: &Tensor<T>, labels: &Tensor<T>, ) -> Result<()>

where T: Clone + Default + Zero + Send + Sync + 'static + NumCast,

Insert tensors into cache

pub fn clear(&mut self) -> Result<()>

Clear cache