我想获取张量中元素的计数，例如t = [1, 2, 0, 0, 0, 0]（t是张量）。我可以通过t.count(0)在Python中调用来获得零的数量4 ，但是在TensorFlow中，我找不到任何函数来执行此操作。如何获得零计数？
TensorFlow目前没有内置的计数方法。但是您可以使用现有的工具，例如：
def tf_count(t, val):
elements_equal_to_value = tf.equal(t, val)
as_ints = tf.cast(elements_equal_to_value, tf.int32)
count = tf.reduce_sum(as_ints)
return count
本文收集自互联网，转载请注明来源。
如有侵权，请联系[email protected] 删除。
我来说两句