比较张量的每个元素的张量

Question

我的输入张量的大小为 3x5。我试图获得每个张量的总数，值超过1。例如：

input list[[0.1 , 1.1 , 1.3, 1.5 , 0.7] ,
           [1.1 , 1.1 , 0.8, 1.5 , 0.7] ,
           [0.1 , 0.0 , 1.3, 0.5 , 1.7]]  
return[[3],[3],[2]]  

因为在列表中，有3、3、2个数字具有大于1的值。

我最初希望使用下面的代码来解决它，

tf.reduce_sum(tf.where(tf.greater(inputs , one),one,zero),1)

'一个'是一个张量，大小为 3x5，所有值均为1。"零"是一个张量，大小为 3x5，所有值零。
但是我意识到不能以这种方式使用tf.greater。有没有建议解决我的问题的代码？我已经陷入了相当长的一段时间，但没有找到解决方案。非常感谢！

Answer 1

此返回

[3 3 2]

import tensorflow as tf
inputlist = [[0.1 , 1.1 , 1.3, 1.5 , 0.7] ,
           [1.1 , 1.1 , 0.8, 1.5 , 0.7] ,
           [0.1 , 0.0 , 1.3, 0.5 , 1.7]]
x = tf.Variable(initial_value=inputlist)
sess = tf.Session()
sess.run(tf.global_variables_initializer())
print( sess.run( tf.count_nonzero(( tf.greater(inputlist, tf.constant(1.0)) ), 1)) )