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, const bool skip_mode = false)
Method to init the op’s attributes.
- 参数
keep_dims – [in] Define whether keep the dims reduced, default false.
skip_mode – [in] Define whether skip reduce, default false.
-
void set_keep_dims(const bool keep_dims)
Method to set keep_dims attribute.
- 参数
keep_dims – [in] Define whether keep the dims reduced, default false.
-
bool get_keep_dims() const
Method to get keep_dims attribute.
- 返回
keep_dims attribute.
-
void set_skip_mode(const bool skip_mode)
Method to set skip_mode attribute.
- 参数
skip_mode – [in] Define whether skip reduce, default false.
-
bool get_skip_mode() const
Method to get skip_mode attribute.
- 返回
skip_mode attribute.
-
inline Reduce()