Class TopK
Defined in File topk.h
Inheritance Relationships
Base Type
public mindspore::ops::BaseOperator
Derived Type
public mindspore::ops::TopKFusion
(Class TopKFusion)
Class Documentation
-
class TopK : public mindspore::ops::BaseOperator
Finds values and indices of the k largest entries along the last dimension. Refer to Python API mindspore.ops.TopK for more details.
Subclassed by mindspore::ops::TopKFusion
Public Functions
-
void Init(const bool sorted = false)
Init. Refer to the parameters of Python API mindspore.ops.TopK for the inputs.
-
void set_sorted(const bool sorted)
Set sorted.
-
bool get_sorted() const
Get sorted.
- Returns
sorted.
-
void Init(const bool sorted = false)