Struct GradientDict 

pub struct GradientDict { /* private fields */ }

Dictionary for storing parameter gradients

Implementations

impl GradientDict

pub fn new() -> Self

Create a new empty gradient dictionary

pub fn insert(&mut self, name: String, gradient: Box<dyn ArrayProtocol>)

Insert a gradient for a parameter

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

Get a gradient by parameter name

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

Get a mutable reference to a gradient by parameter name

pub fn iter(&self) -> impl Iterator<Item = (&String, &Box<dyn ArrayProtocol>)>

Iterate over parameter names and gradients

pub fn merge(&mut self, other: GradientDict)

Merge another gradient dictionary into this one

pub fn is_empty(&self) -> bool

Check if the dictionary is empty

pub fn len(&self) -> usize

Get the number of gradients

pub fn clear(&mut self)

Clear all gradients

pub fn keys(&self) -> impl Iterator<Item = &String>

Get all parameter names

pub fn values(&self) -> impl Iterator<Item = &Box<dyn ArrayProtocol>>

Get all gradients

Trait Implementations

impl Clone for GradientDict

fn clone(&self) -> GradientDict

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for GradientDict

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

Formats the value using the given formatter.

impl Default for GradientDict

fn default() -> Self

Returns the “default value” for a type.