Class Reduce

Inheritance Relationships

Base Type

  • public mindspore::ops::BaseOperator

Derived Type

Class Documentation

class Reduce : public mindspore::ops::BaseOperator

Reduce defined Reduce operator prototype of lite.

Subclassed by mindspore::ops::ReduceFusion

Public Functions

inline Reduce()

Constructor.

inline explicit Reduce(const std::string k_name)

Constructor.

void Init(const bool keep_dims = false)

Method to init the op’s attributes.

Parameters

keep_dims – [in] Define whether keep the dims reduced, default false.

void set_keep_dims(const bool keep_dims)

Method to set keep_dims attribute.

Parameters

keep_dims – [in] Define whether keep the dims reduced, default false.

bool get_keep_dims() const

Method to get keep_dims attribute.

Returns

keep_dims attribute.