来源于解决一个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)))
GitHub 加速计划 / te / tensorflow
184.55 K
74.12 K
下载
一个面向所有人的开源机器学习框架
最近提交(Master分支:2 个月前 )
a49e66f2 PiperOrigin-RevId: 663726708 2 个月前
91dac11a This test overrides disabled_backends, dropping the default value in the process. PiperOrigin-RevId: 663711155 2 个月前
Logo

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

更多推荐