Class Scale

Inheritance Relationships

Base Type

Derived Type

Class Documentation

class Scale : public mindspore::ops::PrimitiveC

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.

~Scale() = default

Destructor.

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.