Class Reduce

• Defined in File reduce.h

Inheritance Relationships

Base Type

• public mindspore::ops::PrimitiveC (Class PrimitiveC)

Derived Types

• public mindspore::ops::ReduceASum (Class ReduceASum)

• public mindspore::ops::ReduceAll (Class ReduceAll)

• public mindspore::ops::ReduceAny (Class ReduceAny)

• public mindspore::ops::ReduceFusion (Class ReduceFusion)

• public mindspore::ops::ReduceMax (Class ReduceMax)

• public mindspore::ops::ReduceMean (Class ReduceMean)

• public mindspore::ops::ReduceMin (Class ReduceMin)

• public mindspore::ops::ReduceProd (Class ReduceProd)

• public mindspore::ops::ReduceSum (Class ReduceSum)

• public mindspore::ops::ReduceSumSquare (Class ReduceSumSquare)

Class Documentation

class Reduce : public mindspore::ops::PrimitiveC

Reduce defined Reduce operator prototype of lite.

Subclassed by mindspore::ops::ReduceASum, mindspore::ops::ReduceAll, mindspore::ops::ReduceAny, mindspore::ops::ReduceFusion, mindspore::ops::ReduceMax, mindspore::ops::ReduceMean, mindspore::ops::ReduceMin, mindspore::ops::ReduceProd, mindspore::ops::ReduceSum, mindspore::ops::ReduceSumSquare

Public Functions

inline Reduce()

Constructor.

inline explicit Reduce(const std::string k_name)

Constructor.

~Reduce() = default

Destructor.

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.