ArgMax  // Argmax

C++ Interface

class ArgMax : 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 ArgMax operation.

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

Constructs a ArgMax operation.

  • arg: The input tensor

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

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