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

pub trait ComputeOutput<T, B: IBackend> {
    fn compute_output(&self, backend: &B, weights_data: &[&SharedTensor<T>], input_data: &[&SharedTensor<T>], output_data: &mut [&mut SharedTensor<T>]);
}

A Layer that can compute the output for a given input.

Required Methods

fn compute_output(&self, backend: &B, weights_data: &[&SharedTensor<T>], input_data: &[&SharedTensor<T>], output_data: &mut [&mut SharedTensor<T>])

Compute output for given input and write them into output_data.

Implementors