我正在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。但是我不知道如何使用它。
最佳答案
要回答我自己的问题,C++中没有直接的方法可以完成这项工作。
解决方法是迭代获取和存储每个输出值,并获取列表的最大值。
关于c++ - TensorFlow C++等于argmax(axis = -1),我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/54854890/