ArgMin  // Argmin

C++ Interface

C++ Interface

class ArgMin : public ngraph::op::util::IndexReduction

Computes minimum index along a specified axis for a given tensor.

Public Functions

const NodeTypeInfo &get_type_info() const

Returns the NodeTypeInfo for the node’s class. During transition to type_info, returns a dummy type_info for Node if the class has not been updated yet.


Constructs a ArgMin operation.

ArgMin(const Output<Node> &arg, size_t axis, const element::Type &index_element_type)

Constructs a ArgMin operation.

  • arg: The input tensor

  • axis: The axis along which to compute an index for minimum

  • index_element_type: produce indices. Currently, only int64 or int32 are supported