Skip to main content

Cost

Trait Cost 

Source
pub trait Cost<B: Backend> {
    // Required methods
    fn loss<D: Dimension + RemoveAxis>(
        &self,
        predicted: &B::Tensor<D>,
        target: &B::Tensor<D>,
    ) -> f32;
    fn gradient<D: Dimension>(
        &self,
        predicted: &B::Tensor<D>,
        target: &B::Tensor<D>,
    ) -> B::Tensor<D>;
}

Required Methods§

Source

fn loss<D: Dimension + RemoveAxis>( &self, predicted: &B::Tensor<D>, target: &B::Tensor<D>, ) -> f32

Source

fn gradient<D: Dimension>( &self, predicted: &B::Tensor<D>, target: &B::Tensor<D>, ) -> B::Tensor<D>

Dyn Compatibility§

This trait is not dyn compatible.

In older versions of Rust, dyn compatibility was called "object safety", so this trait is not object safe.

Implementors§