以布尔十位数表示的“真”值的计数

Question

我知道tf.where将返回True值的位置，这样我就可以使用结果的shape[0]来获得True的数量

但是，当我尝试使用它时，维度是未知的（这是有意义的，因为它需要在运行时计算）。所以我的问题是，我怎样才能访问一个维度，并在一个操作中像求和一样使用它？

例如：

myOtherTensor = tf.constant([[True, True], [False, True]])
myTensor = tf.where(myOtherTensor)
myTensor.get_shape() #=> [None, 2]
sum = 0
sum += myTensor.get_shape().as_list()[0] # Well defined at runtime but considered None until then.

Answer 1

您可以将值强制转换为浮点数并计算它们的和： tf.reduce_sum(tf.cast(myOtherTensor, tf.float32))

根据实际的用例，如果指定了调用的reduce维度，还可以计算每行/列的总和。