Skip to main content

LossFunction

tml_utils::network

Trait LossFunction 

Source 
pub trait LossFunction<const N: usize>: Debug {
    // Required method
    fn loss_and_grad(
        &self,
        output: &[Float; N],
        target: &[Float; N],
        grad: &mut [Float; N],
    ) -> Float;
}

Required Methods§

Source

fn loss_and_grad( &self, output: &[Float; N], target: &[Float; N], grad: &mut [Float; N], ) -> Float

Implementors§