Struct NpzTensors

pub struct NpzTensors { /* private fields */ }

Lazy tensor loader.

Implementations

impl NpzTensors

pub fn new<T: AsRef<Path>>(path: T) -> Result<Self>

pub fn names(&self) -> Vec<&String>

pub fn get_shape_and_dtype(&self, name: &str) -> Result<(Shape, DType)>

This only returns the shape and dtype for a named tensor. Compared to get, this avoids reading the whole tensor data.

pub fn get(&self, name: &str) -> Result<Option<Tensor>>

Trait Implementations

impl SimpleBackend for NpzTensors

fn get( &self, s: Shape, name: &str, _: Init, dtype: DType, dev: &Device, ) -> Result<Tensor>

Retrieve a tensor based on a target name and shape.

fn get_unchecked( &self, name: &str, dtype: DType, dev: &Device, ) -> Result<Tensor>

Retrieve a tensor based on the name.

fn contains_tensor(&self, name: &str) -> bool