在Tensorflow中,如何在几个张量中选择最小张量?

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

min = tf.minimum(x, y)

但是,我想知道如何在一组张量中选择最小张量。是否有张量中的最小值或最大值之类的函数?


回首忆惘然
浏览 327回答 1
1回答

慕娘9325324

您可以使用stack所有张量,然后使用reduce_mina = 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]
打开App,查看更多内容
随时随地看视频慕课网APP

相关分类

Python