Aily_ran 2018-03-08 13:04 采纳率: 0%
浏览 2938
已结题

Tensorflow 判断两个tensor含有多少一样的元素

例如a=[1,2,3,4,5];b=[3,4,5,6,7]
result = 3(共同含有3,4,5这三个元素)

  • 写回答

1条回答 默认 最新

  • yjjhwander 2018-03-08 13:33
    关注
    import tensorflow as tf
    a = tf.constant([1,2,3,4,5])
    b = tf.constant([3,4,5,6,7])
    c = tf.constant(value=0, dtype=tf.int32)
    for i in tf.unstack(a):
        for j in tf.unstack(b):
            c = tf.cond(tf.equal(i, j), lambda : tf.add(c, 1), lambda : c)
    
    sess = tf.InteractiveSession()
    print(sess.run(c))
    

    求C币,急用。

    评论

报告相同问题?