在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]