ONE - On-device Neural Engine
All Data Structures Namespaces Files Functions Variables Typedefs Enumerations Enumerator Friends Macros Modules Pages
losses.cce.CategoricalCrossentropy Class Reference
Collaboration diagram for losses.cce.CategoricalCrossentropy:

Public Member Functions

 __init__ (self, reduction="mean")
 

Additional Inherited Members

- Data Fields inherited from losses.loss.LossFunction
 reduction
 

Detailed Description

Categorical Cross-Entropy Loss Function with reduction type.

Definition at line 5 of file cce.py.

Constructor & Destructor Documentation

◆ __init__()

losses.cce.CategoricalCrossentropy.__init__ (   self,
  reduction = "mean" 
)
Initialize the Categorical Cross-Entropy loss function.
Args:
    reduction (str): Reduction type ('mean', 'sum').

Reimplemented from losses.loss.LossFunction.

Definition at line 9 of file cce.py.

9 def __init__(self, reduction="mean"):
10 """
11 Initialize the Categorical Cross-Entropy loss function.
12 Args:
13 reduction (str): Reduction type ('mean', 'sum').
14 """
15 super().__init__(reduction)

References losses.cce.CategoricalCrossentropy.__init__().

Referenced by losses.cce.CategoricalCrossentropy.__init__().


The documentation for this class was generated from the following file: