Class Reduce
Defined in File reduce.h
Inheritance Relationships
Base Type
public mindspore::ops::BaseOperator
Derived Type
public mindspore::ops::ReduceFusion
(Class ReduceFusion)
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.
-
inline Reduce()