Struct CustomGradientRegistry 

pub struct CustomGradientRegistry<T, E> { /* private fields */ }

Registry for custom gradient functions

Implementations

impl<T, E> CustomGradientRegistry<T, E>

pub fn new() -> Self

pub fn register<F>(&mut self, operation_name: String, backward_fn: F)

where F: Fn(&T, &[T]) -> Result<Vec<T>, E> + 'static,

Register a custom backward function for an operation

pub fn get(&self, operation_name: &str) -> Option<&BackwardFn<T, E>>

Get custom gradient function for an operation

pub fn has_custom_gradient(&self, operation_name: &str) -> bool

Check if custom gradient exists

pub fn unregister(&mut self, operation_name: &str) -> bool

Remove custom gradient

pub fn len(&self) -> usize

Get number of registered gradients

pub fn is_empty(&self) -> bool

Trait Implementations

impl<T, E> Default for CustomGradientRegistry<T, E>

fn default() -> Self

Returns the “default value” for a type.