tensorflow判断两个tensor是否完全相同

来源:互联网 发布:yymodel 数组转模型 编辑:程序博客网 时间:2024/05/01 22:56

来源于解决一个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)))
阅读全文
0 0