Class ExpFusion

Inheritance Relationships

Base Type

  • public mindspore::ops::BaseOperator

Class Documentation

class ExpFusion : public mindspore::ops::BaseOperator

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.

参数
  • 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.

参数

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.

参数

scale[in] Define a size factor of input.

void set_shift(const float shift)

Method to set shift attribute. Default is 0.0.

参数

shift[in] Define a bias of input.

float get_base() const

Method to get base attribute.

返回

base number.

float get_scale() const

Method to get scale attribute.

返回

a size factor of input.

float get_shift() const

Method to get shift attribute.

返回

a bias of input.