ArgMax

tensorflow C++ API

tensorflow::ops::ArgMax

Returns the index with the largest value across dimensions of a tensor.

Summary

Note that in case of ties the identity of the return value is not guaranteed.

Arguments:

• scope: A Scope object
• dimension: int32, 0 <= dimension < rank(input). Describes which dimension of the input Tensor to reduce across. For vectors, use dimension = 0.

Returns:

• Output: The output tensor.

Constructor

• ArgMax(const ::tensorflow::Scope & scope, ::tensorflow::Input input,::tensorflow::Input dimension).

Public attributes

• tensorflow::Output output.

ArgMax

Source link : https://github.com/EXPNUNI/enuSpaceTensorflow/blob/master/enuSpaceTensorflow/tf_math.cpp

Argument:

• Scope scope : A Scope object (A scope is generated automatically each page. A scope is not connected.).
• Input input: connect Input node.
• Input dimension: connect Input node.

Return:

• Output output: Output object of ArgMax class object.

Result:

• std::vector(Tensor) product_result : Returned object of executed result by calling session.

Using Method