Struct CrossEntropyLoss 

pub struct CrossEntropyLoss { /* private fields */ }

Cross-entropy loss.

Implementations

impl CrossEntropyLoss

pub fn new(reduction: Option<&str>) -> Self

Create a new cross-entropy loss.

Trait Implementations

impl Loss for CrossEntropyLoss

fn forward( &self, predictions: &dyn ArrayProtocol, targets: &dyn ArrayProtocol, ) -> CoreResult<Box<dyn ArrayProtocol>>

Compute the loss between predictions and targets.

fn backward( &self, predictions: &dyn ArrayProtocol, targets: &dyn ArrayProtocol, ) -> CoreResult<Box<dyn ArrayProtocol>>

Compute the gradient of the loss with respect to predictions.

fn name(&self) -> &str

Get the name of the loss function.