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