Struct Serializer

pub struct Serializer {
    pub state: FxHashMap<String, NodeIndex>,
    /* private fields */
}

Serializer keeps track of the tensors and modules that make up a model

Fields

state: FxHashMap<String, NodeIndex>

Implementations

impl Serializer

pub fn tensor<S: Shape>(&mut self, name: &str, tensor: GraphTensor<S>)

pub fn module<T: SerializeModule>(&mut self, name: &str, module: &T)

Trait Implementations

impl Debug for Serializer

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

Formats the value using the given formatter.

impl Default for Serializer

fn default() -> Serializer

Returns the “default value” for a type.