Class Scale
Defined in File scale.h
Inheritance Relationships
Base Type
public mindspore::ops::BaseOperator
Derived Type
public mindspore::ops::ScaleFusion
(Class ScaleFusion)
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.
- Parameters
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.
- Parameters
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.
- Returns
axis attribute.
-
inline Scale()