C ++ TensorFlow:如何在“Mean”中指定“axis”

时间:2017-07-17 17:18:06

标签: c++ machine-learning tensorflow deep-learning

我正在尝试在C ++ TensorFlow中实现一个简单的前馈神经网络。我无法在C ++ tensorflow(https://www.tensorflow.org/api_docs/cc/class/tensorflow/ops/mean)的“Mean”函数中指定“axis”。

我正在尝试在C ++ TensorFlow中实现以下Python-TensorFlow行:

cost = tf.reduce_mean(tf.nn.softmax_cross_entropy_with_logits(logits=pred, labels=y))

以下是在C ++ TensorFlow中,这就是我想要的内容相当于上面的行(在Python TensorFlow中)

auto cost = Mean(root.WithOpName("cost"), SoftmaxCrossEntropyWithLogits(root, output_layer, y));

我收到类似“指定轴”的错误,我不知道如何在上面的C ++ TensorFlow代码中指定“axis”。我在网上搜索但一无所获。网上没有例子,甚至在官方文档中都没有。请帮帮我......

0 个答案:

没有答案