Struct StateDict 

pub struct StateDict {
    pub tensors: HashMap<String, Tensor>,
}

PyTorch-like state dict: a HashMap<String, Tensor>.

Fields

tensors: HashMap<String, Tensor>

The underlying key-value store.

Implementations

impl StateDict

pub fn new() -> Self

Create an empty state dict.

pub fn insert(&mut self, key: impl Into<String>, tensor: Tensor)

Insert a tensor under a key.

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

Retrieve a tensor by key.

pub fn len(&self) -> usize

Number of tensors.

pub fn is_empty(&self) -> bool

Whether the dict is empty.

pub fn total_params(&self) -> usize

Total number of parameters (sum of all tensor element counts).

pub fn to_bytes(&self) -> Vec<u8>

Serialise to bytes.

Format: \[entry_count: u64\](\[key_len: u64\][key_bytes]\[tensor_len: u64\][tensor_bytes])...

pub fn from_bytes(bytes: &[u8]) -> Option<Self>

Deserialise from bytes produced by StateDict::to_bytes.

Trait Implementations

impl Clone for StateDict

fn clone(&self) -> StateDict

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for StateDict

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

Formats the value using the given formatter.

impl Default for StateDict

fn default() -> StateDict

Returns the “default value” for a type.