计算布尔张量中“True”值的数量

10 浏览
0 Comments

计算布尔张量中“True”值的数量

我知道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] # 在运行时定义明确但在那之前被视为None。

0