Class Eltwise

Inheritance Relationships

Base Type

Class Documentation

class Eltwise : public mindspore::ops::PrimitiveC

Eltwise defined Element-wise operator prototype.

Public Functions

inline Eltwise()

Constructor.

~Eltwise() = default

Destructor.

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.