Class ExpFusion
Defined in File exp_fusion.h
Inheritance Relationships
Base Type
public mindspore::ops::Exp
(Class Exp)
Class Documentation
-
class ExpFusion : public mindspore::ops::Exp
ExpFusion defined Exp operator prototype of lite.
Public Functions
-
inline ExpFusion()
Constructor.
-
void Init(const float base = -1.0, const float scale = 1.0, const float shift = 0.0)
Method to init the op’s attributes.
- Parameters
base – [in] Define a base number. If base is -1, it represents e. In addition to this, base must be larger than 0.
scale – [in] Define a size factor of input.
shift – [in] Define a bias of input.
-
void set_base(const float base)
Method to set base attribute.
- Parameters
base – [in] Define a base number. If base is -1, it represents e. In addition to this, base must be larger than 0. Default is -1.
-
void set_scale(const float scale)
Method to set scale attribute. Default is 1.0.
- Parameters
scale – [in] Define a size factor of input.
-
void set_shift(const float shift)
Method to set shift attribute. Default is 0.0.
- Parameters
shift – [in] Define a bias of input.
-
float get_base() const
Method to get base attribute.
- Returns
base number.
-
float get_scale() const
Method to get scale attribute.
- Returns
a size factor of input.
-
float get_shift() const
Method to get shift attribute.
- Returns
a bias of input.
-
inline ExpFusion()