TensorFlow C ++等于argmax（axis = -1）

Question

我正在预测TensorFlow C ++中的pb图输出。 会话->运行正常，并给出浮点值列表作为输出

load_graph_status = session->Run(inputs, { output_layer_name }, {}, &outputs);

我在使用过的Python中做了类似的预测

output = outputs.argmax(axis=-1)

我在C ++中找不到与之等效的东西？ TensorFlow C ++文档中有一个tensorflow :: ops :: argmax。但是我不知道如何使用它。

Answer 1

要回答我自己的问题，C ++中没有直接的方法可以完成这项工作。 解决方法是迭代获取和存储每个输出值，并获取列表的最大值。