Trait CrossEntropy

pub trait CrossEntropy {
    type Output;

    // Required method
    fn cross_entropy(&self) -> Self::Output;
}

A trait for computing the cross-entropy loss of a tensor or array

Required Associated Types

type Output

Required Methods

fn cross_entropy(&self) -> Self::Output

Implementations on Foreign Types

impl<A, S, D> CrossEntropy for ArrayBase<S, D>

where A: Float + FromPrimitive + ScalarOperand, D: Dimension, S: Data<Elem = A>,

type Output = A

fn cross_entropy(&self) -> Self::Output

Implementors