tensorflow比较张量的每个元素

Question

我的输入Tensor的大小为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)

'one'是尺寸为3x5的张量，所有值均为1.'零'是尺寸为3x5的张量，所有值均为零。
但我意识到tf.greater不能以这种方式使用。有没有推荐的代码来解决我的问题？我一直坚持这个问题很长一段时间没有找到解决方案。非常感谢！

Answer 1

这将返回

  

[3 3 2]

1535547714000