Class Scale

Inheritance Relationships

Base Type

  • public mindspore::ops::BaseOperator

Derived Type

Class Documentation

class Scale : public mindspore::ops::BaseOperator

Scale defined Scale operator prototype of lite.

Subclassed by mindspore::ops::ScaleFusion

Public Functions

inline Scale()

Constructor.

inline explicit Scale(const std::string k_name)

Constructor.

void Init(const int64_t axis = -1)

Method to init the op's attributes.

参数

axis – [in] Define the first axis of input[0] along which to apply input[1], can be negative to index from the end. Default -1.

void set_axis(const int64_t axis)

Method to set axis attribute.

参数

axis – [in] Define the first axis of input[0] along which to apply input[1], can be negative to index from the end. Default -1.

int64_t get_axis() const

Method to get axis attribute.

返回

axis attribute.