How to calculate the Cosine similarity between two tensors?

前端 未结 3 908
無奈伤痛
無奈伤痛 2020-12-08 11:29

I have two normalized tensors and I need to calculate the cosine similarity between these tensors. How do I do it with TensorFlow?

cosine(normalize_a,normali         


        
3条回答
  •  暗喜
    暗喜 (楼主)
    2020-12-08 12:09

    This will do the job:

    a = tf.placeholder(tf.float32, shape=[None], name="input_placeholder_a")
    b = tf.placeholder(tf.float32, shape=[None], name="input_placeholder_b")
    normalize_a = tf.nn.l2_normalize(a,0)        
    normalize_b = tf.nn.l2_normalize(b,0)
    cos_similarity=tf.reduce_sum(tf.multiply(normalize_a,normalize_b))
    sess=tf.Session()
    cos_sim=sess.run(cos_similarity,feed_dict={a:[1,2,3],b:[2,4,6]})
    

    This prints 0.99999988

提交回复
热议问题