在Tensorflow中，如何在幾個張量中選擇最小張量？

Question

根據Tensorflow API，我知道我可以沿着特定的軸reduce_min或reduce_max。 我還可以如下選擇兩個張量中的最小張量。

min = tf.minimum(x, y)

但是，我想知道如何在一組張量中選擇最小張量。 是否有張量中的最小值或最大值之類的函數？

非常感謝！

Answer 1

您可以stack所有張量，然后使用reduce_min

a = tf.constant([3])
b = tf.constant([2])
c = tf.constant([1])

with tf.Session() as sess:
   print(tf.reduce_min(tf.stack([a,b,c]),0).eval())
#[1]