Class ArgMin
Defined in File arg_min.h
Inheritance Relationships
Base Type
public mindspore::ops::BaseOperator
Derived Type
public mindspore::ops::ArgMinFusion
(Class ArgMinFusion)
Class Documentation
-
class ArgMin : public mindspore::ops::BaseOperator
Returns the indices of the minimum value of a tensor across the axis. Refer to Python API mindspore.ops.Argmin for more details.
Subclassed by mindspore::ops::ArgMinFusion
Public Functions
-
inline ArgMin()
Constructor.
-
void Init(const int64_t axis = -1, const TypeId output_type = kNumberTypeInt32)
Init. Refer to the parameters of Python API mindspore.ops.Argmin 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.
- Returns
axis.
-
TypeId get_output_type() const
Get output_type.
- Returns
output_type.
-
inline ArgMin()