pub trait StepOptimizer: Sealed {
// Required method
fn step(
&mut self,
parameter_id: u64,
weights: &mut Tensor,
grad: &Tensor,
) -> Result<(), OptimError>;
}Expand description
Trait for optimizers that support a per-parameter step update.
This trait is sealed and cannot be implemented outside this crate.
Required Methods§
fn step( &mut self, parameter_id: u64, weights: &mut Tensor, grad: &Tensor, ) -> Result<(), OptimError>
Dyn Compatibility§
This trait is dyn compatible.
In older versions of Rust, dyn compatibility was called "object safety".