tensorflow判断两个tensor是否完全相同
tensorflow
一个面向所有人的开源机器学习框架
项目地址:https://gitcode.com/gh_mirrors/te/tensorflow
免费下载资源
·
来源于解决一个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 个月前
更多推荐
已为社区贡献10条内容
所有评论(0)