Struct auto_diff::op::loss::MSELoss

pub struct MSELoss { /* private fields */ }

MSELoss The left-most dimension is the N.

Implementations

impl MSELoss

pub fn new() -> MSELoss

Trait Implementations

impl OpTrait for MSELoss

fn get_name(&self) -> String

A conventional name for the op

fn get_input_size(&self) -> usize

The number of input needs by this op.

fn get_output_size(&self) -> usize

The number of output produced by this op.

fn apply(&self, input: &[Tensor], output: &[Tensor])

Forward pass

fn grad(&self, input: &[Tensor], output_grad: &[Tensor], input_grad: &[Tensor])

Given the forward input value and backward output_grad, Update weight gradient. return backward input gradeint.

fn get_values(&self) -> Vec<Tensor>

access weight values

fn set_values(&self, _v: &[Tensor])

fn get_grads(&self) -> Vec<Tensor>

access gradient values