pub trait AutodiffModel: Model {
// Provided methods
fn forward_autodiff(
&self,
input: &ArrayView<'_, f64, Ix2>,
) -> TrainResult<()> { ... }
fn compute_gradients(&self) -> TrainResult<HashMap<String, Array<f64, Ix2>>> { ... }
}Expand description
Trait for models that support automatic differentiation via scirs2-autograd.
This trait extends the base Model trait with support for training using SciRS2’s automatic differentiation system.
Note: This trait is currently a placeholder for future scirs2-autograd integration. The actual Variable type will be specified once scirs2-autograd is fully integrated.
Provided Methods§
Sourcefn forward_autodiff(&self, input: &ArrayView<'_, f64, Ix2>) -> TrainResult<()>
fn forward_autodiff(&self, input: &ArrayView<'_, f64, Ix2>) -> TrainResult<()>
Sourcefn compute_gradients(&self) -> TrainResult<HashMap<String, Array<f64, Ix2>>>
fn compute_gradients(&self) -> TrainResult<HashMap<String, Array<f64, Ix2>>>
Compute gradients automatically using backward pass (placeholder).
§Returns
Gradients for all parameters