在Tensorflow中，如何在几个张量中选择最小张量？

Question

根据Tensorflow API，我知道我可以沿着特定的轴reduce_min或reduce_max。我还可以从以下两个张量中选择最小张量。

min = tf.minimum(x, y)

但是，我想知道如何在一组张量中选择最小张量。张量中是否存在像min或max这样的函数？

非常感谢您！

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]