Trait TaskHead 

pub trait TaskHead: Debug {
    // Required methods
    fn forward(&self, features: &Array4<f64>) -> Result<TaskOutput>;
    fn parameters(&self) -> &Array1<f64>;
    fn update_parameters(&mut self, params: &Array1<f64>) -> Result<()>;
    fn clone_box(&self) -> Box<dyn TaskHead>;
}

Trait for task-specific heads

Required Methods

fn forward(&self, features: &Array4<f64>) -> Result<TaskOutput>

Process features for specific task

fn parameters(&self) -> &Array1<f64>

Get head parameters

fn update_parameters(&mut self, params: &Array1<f64>) -> Result<()>

Update parameters

fn clone_box(&self) -> Box<dyn TaskHead>

Clone the head

Trait Implementations

impl Clone for Box<dyn TaskHead>

fn clone(&self) -> Self

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

Implementors