TensorFlow相当于numpy.all()

Bru*_* KM 4 tensorflow

如标题中所述,是否有一个TensorFlow等效的numpy.all()函数来检查bool张量中的所有值是否都是True?实施此类检查的最佳方法是什么?

Mir*_*ber 5

使用tf.reduce_all,如下所示:

import tensorflow as tf
a=tf.constant([True,False,True,True],dtype=tf.bool)
res=tf.reduce_all(a)
sess=tf.InteractiveSession()
res.eval()
Run Code Online (Sandbox Code Playgroud)

这回来了False.

另一方面,这会返回True:

import tensorflow as tf
a=tf.constant([True,True,True,True],dtype=tf.bool)
res=tf.reduce_all(a)
sess=tf.InteractiveSession()
res.eval()
Run Code Online (Sandbox Code Playgroud)