Trait juice::layer::ComputeOutput

source · 
pub trait ComputeOutput<T, B: IBackend> {
    // Required method
    fn compute_output(
        &self,
        backend: &B,
        weights_data: &[&SharedTensor<T>],
        input_data: &[&SharedTensor<T>],
        output_data: &mut [&mut SharedTensor<T>]
    );
}
Expand description

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

Required Methods§

source

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§