Struct NetworkRegistry 

pub struct NetworkRegistry { /* private fields */ }

Registry for managing neural networks.

The registry maintains a collection of NetworkHandle instances, each identified by a unique name. Networks are registered with configurations and then have their PyTorch modules attached via the Python API.

Implementations

impl NetworkRegistry

pub fn new() -> Self

Create a new empty registry.

pub fn register(&mut self, config: NetworkConfig)

Register a network with the given configuration.

If a network with the same name already exists, it will be replaced.

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

Get a reference to a network handle by name.

pub fn get_mut(&mut self, name: &str) -> Option<&mut NetworkHandle>

Get a mutable reference to a network handle by name.

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

Check if a network is registered.

pub fn unregister(&mut self, name: &str) -> Option<NetworkHandle>

Remove a network from the registry.

pub fn set_train_mode(&mut self, train: bool)

Set train mode for all registered networks.

This affects both the train_mode flag on handles and should be used to call .train() or .eval() on PyTorch modules.

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

Get the names of all registered networks.

pub fn len(&self) -> usize

Get the number of registered networks.

pub fn is_empty(&self) -> bool

Check if the registry is empty.

pub fn clear(&mut self)

Remove all networks from the registry.

pub fn iter(&self) -> impl Iterator<Item = (&str, &NetworkHandle)>

Iterate over all network handles.

pub fn iter_mut(&mut self) -> impl Iterator<Item = (&str, &mut NetworkHandle)>

Iterate mutably over all network handles.

pub fn register_embedding(&mut self, handle: EmbeddingHandle)

Register an embedding with the given handle.

pub fn get_embedding(&self, name: &str) -> Option<&EmbeddingHandle>

Get a reference to an embedding handle by name.

pub fn get_embedding_mut(&mut self, name: &str) -> Option<&mut EmbeddingHandle>

Get a mutable reference to an embedding handle by name.

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

Check if an embedding is registered.

Trait Implementations

impl Default for NetworkRegistry

fn default() -> Self

Returns the “default value” for a type.