Trait leaf::layer::ComputeParametersGradient [−] [src]

pub trait ComputeParametersGradient<T, B: IBackend> {
    fn compute_parameters_gradient(&self, backend: &B, output_data: &[&SharedTensor<T>], output_gradients: &[&SharedTensor<T>], input_data: &[&SharedTensor<T>], parameters_gradients: &mut [&mut SharedTensor<T>]) { ... }
}

A Layer that can compute the gradient with respect to its parameters (= weights, bias, etc.).

Provided Methods

fn compute_parameters_gradient(&self, backend: &B, output_data: &[&SharedTensor<T>], output_gradients: &[&SharedTensor<T>], input_data: &[&SharedTensor<T>], parameters_gradients: &mut [&mut SharedTensor<T>])

Compute gradients with respect to the parameters and write them into parameters_gradients.

Implementors