Class Eltwise

• Defined in File eltwise.h

Inheritance Relationships

Base Type

• public mindspore::ops::BaseOperator

Class Documentation

class Eltwise : public mindspore::ops::BaseOperator

Eltwise defined Element-wise operator prototype.

Public Functions

inline Eltwise()

Constructor.

void Init(const EltwiseMode &mode)

Method to init the op’s attributes.

Parameters

mode – [in] Define the concrete computation operator.

void set_mode(const EltwiseMode &mode)

Method to set mode attribute.

Parameters

mode – [in] Define the concrete computation operator.

EltwiseMode get_mode() const

Method to get mode attribute.

Returns

an enumerated value to indicate the concrete computation operator.