来源于解决一个csdn中的问答,问答链接:http://ask.csdn.net/questions/652382#answer_460706

笔者想到的方法就是直接把bool类型转换到int处理,之后方法就很多样了(求和、求最小值等等,总归是利用一些全一序列的性质)。下面给一个例子:

 a = tf.placeholder(tf.int32,[2,])
 b = tf.placeholder(tf.int32,[2,])
 aeqb = tf.equal(a,b)
 aeqb_int = tf.to_int32(aeqb)
 result = tf.equal(tf.reduce_sum(aeqb_int),tf.reduce_sum(tf.ones_like(aeqb_int)))
Logo

旨在为数千万中国开发者提供一个无缝且高效的云端环境,以支持学习、使用和贡献开源项目。

更多推荐