Class Argmax

Defined in File arg_max.h

Inheritance Relationships

Base Type

public mindspore::ops::BaseOperator

Derived Type

public mindspore::ops::ArgMaxFusion (Class ArgMaxFusion)

Class Documentation

class Argmax : public mindspore::ops::BaseOperator 

Returns the indices of the maximum value of a tensor across the axis. Refer to Python API mindspore.ops.Argmax for more details.

Subclassed by mindspore::ops::ArgMaxFusion

Public Functions

inline Argmax(): Constructor.

void Init(const int64_t axis = -1, const TypeId output_type = kNumberTypeInt32): Init. Refer to the parameters of Python API mindspore.ops.Argmax for the inputs.

void set_axis(const int64_t axis): Set axis.

void set_output_type(const TypeId output_type): Set output_type.

int64_t get_axis() const

Get axis.

返回: axis.

TypeId get_output_type() const

Get output_type.

返回: output_type.