如何计算向量和三维张量之间的欧氏距离？

作者: 红蜻蜓
发布时间: 2024-12-11 11:20:42 (18天前)
转自：

2 条回复

0#
回复此人
無口君 | 2019-08-31 10-32

<div class =“post-text”itemprop =“text”> <P> 你可以这样做 <a href="https://www.tensorflow.org/api_docs/python/tf/norm" rel="nofollow noreferrer"> <code> tf.norm </code> </A> 和 <a href="https://www.tensorflow.org/api_docs/python/tf/reduce_mean" rel="nofollow noreferrer"> <code> tf.reduce_mean </code> </A> ： </p> <pre> <code> import tensorflow as tf with tf.Graph().as_default(), tf.Session() as sess: a = tf.placeholder(tf.float32, [1, None]) b = tf.placeholder(tf.float32, [None, None, None]) dist = tf.reduce_mean(tf.norm(b - a, axis=2), axis=1) print(sess.run(dist, feed_dict={a: [[1, 2, 3]], b: [[[ 4, 5, 6], [ 7, 8, 9]], [[10, 11, 12], [13, 14, 15]]]})) # [ 7.7942286 18.186533 ] </code> </pre> </DIV>

编辑

登录后才能参与评论