Class SmoothL1Loss
Defined in File smooth_l1_loss.h
Inheritance Relationships
Base Type
public mindspore::ops::BaseOperator
Class Documentation
-
class SmoothL1Loss : public mindspore::ops::BaseOperator
Computes smooth L1 loss, a robust L1 loss. Refer to Python API mindspore.ops.SmoothL1Loss for more details.
Public Functions
-
inline SmoothL1Loss()
Constructor.
-
void Init(const float beta, const std::string reduction = "none")
Init. Refer to the parameters of Python API mindspore.ops.SmoothL1Loss for the inputs.
-
void set_beta(const float beta)
Set beta.
-
float get_beta() const
Get beta.
- Returns
beta.
-
void set_reduction(const std::string reduction)
Set reduction.
-
std::string get_reduction() const
Get reduction.
- Returns
reduction.
-
inline SmoothL1Loss()