c++ - TensorFlow C++等于argmax(axis = -1)

Question

我正在TensorFlow C++中预测PB图形输出。
session-> Run正常运行，并给出浮点值列表作为输出

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++中没有直接的方法可以完成这项工作。
解决方法是迭代获取和存储每个输出值，并获取列表的最大值。