Struct MSELoss 

pub struct MSELoss { /* private fields */ }

Mean squared error loss.

Implementations

impl MSELoss

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

Create a new MSE loss.

Trait Implementations

impl Loss for MSELoss

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.