Struct candle_core::npy::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>>