如何在TensorFlow C API(例如Python）中设置轴

How to set axis in tensorflow C++ API such as python

本文关键字：Python 设置 例如 TensorFlow API      更新时间：2023-10-16

我在python中使用tensorflow轻松用于数学操作，例如redus_sum或shied_mean 

array = np.ndarray(shape=(2, 2, 3), buffer=np.array([[[1, 2, 3], [4, 5, 6]],
                                                     [[7, 8, 9], [10, 11, 12]]]),
                   dtype=int)
mean = tf.reduce_mean(array)
sum = tf.reduce_sum(array)
with tf.Session() as sess:
    print(sess.run(mean))
    print(sess.run(sum))

从中，我可以得到张量的平均值和总和为一个值，howerver，当我在C 中进行这些操作时，我会遇到一些问题，例如这个问题

Sum(root.WithOpName("sum"), tensor_input, 1)

在此示例中，第二个param Tensor_input是形状的张量[1，160，160，3]。不同的是，我必须将第三个参数设置为(-rank，等级(范围内的数字，但这无法获得我所需的结果，以使张量中的所有值(例如python中的所有值(，而不是，它计算张张量的元素。因此，如何获得与python相同的结果，以将所有值加成一个值。如果有人可以帮助我

，这将有所帮助

我已经解决了它，当您要减少总和或平均结果时，如果您在shapr中的张量[1，160，160，3]中进行张量，则可以使用这个

Sum(root.WithOpName("sum"), tensor_input, {0, 1, 2, 3})

最后一个prama是(0，等级(tensor_input((

相关文章：