Struct NllLoss

pub struct NllLoss {
    pub axis: i32,
    pub reduction: LossReduction,
}

Computes the negative log likelihood loss.

Fields

axis: i32

distribution axis. Default to NllLoss::DEFAULT_AXIS

reduction: LossReduction

Reduction type. Default to NllLoss::DEFAULT_REDUCTION

Implementations

impl NllLoss

pub fn new() -> Self

Creates a new instance of NllLoss.

impl NllLoss

pub const DEFAULT_AXIS: i32 = -1i32

Default value for the axis parameter.

pub const DEFAULT_REDUCTION: LossReduction = LossReduction::None

Default value for the reduction parameter.

pub fn apply( &self, inputs: impl AsRef<Array>, targets: impl AsRef<Array>, ) -> Result<Array, Exception>

Compute the negative log likelihood loss.

Params

• inputs: predicted distribution in log space
• targets: target values

Trait Implementations

impl Buildable for NllLoss

type Builder = NllLossBuilder

The builder type for this buildable type

impl Builder<NllLoss> for NllLossBuilder

type Error = Infallible

Error with building

fn build(self) -> Result<NllLoss, Self::Error>

Build the type

impl Clone for NllLoss

fn clone(&self) -> NllLoss

Returns a copy of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl Debug for NllLoss

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.