ONE - On-device Neural Engine
|
Public Member Functions | |
__init__ (self, reduction="mean") | |
Additional Inherited Members | |
![]() | |
reduction | |
losses.mse.MeanSquaredError.__init__ | ( | self, | |
reduction = "mean" |
|||
) |
Initialize the MSE loss function. Args: reduction (str): Reduction type ('mean', 'sum').
Reimplemented from losses.loss.LossFunction.
Definition at line 9 of file mse.py.
References losses.mse.MeanSquaredError.__init__().
Referenced by losses.mse.MeanSquaredError.__init__().