计算2个张量所有元素之间的距离

作者: 圈圈红
发布时间: 2024-12-17 05:23:49 (2月前)
转自：

3 条回复

0#
回复此人
解天 | 2019-08-31 10-32

<div class =“post-text”itemprop =“text”> <P> 我认为这样做你想要的： </p> <pre> <code> import tensorflow as tf def all_distances(a, b): dists = tf.expand_dims(tf.reshape(a, [-1]), 1) - tf.reshape(b, [-1]) return tf.reshape(dists, tf.concat([tf.shape(a), tf.shape(b)], axis=0)) with tf.Graph().as_default(), tf.Session() as sess: a = tf.constant([[1, 2, 3], [4, 5, 6], [7, 8, 9]]) b = tf.constant([[11, 12], [13, 14], [15, 16]]) dists = all_distances(a, b) print(sess.run(dists)) </code> </pre> <P> 输出： </p> <pre class="lang-none prettyprint-override"> <code> [[[[-10 -11] [-12 -13] [-14 -15]] [[ -9 -10] [-11 -12] [-13 -14]] [[ -8 -9] [-10 -11] [-12 -13]]] [[[ -7 -8] [ -9 -10] [-11 -12]] [[ -6 -7] [ -8 -9] [-10 -11]] [[ -5 -6] [ -7 -8] [ -9 -10]]] [[[ -4 -5] [ -6 -7] [ -8 -9]] [[ -3 -4] [ -5 -6] [ -7 -8]] [[ -2 -3] [ -4 -5] [ -6 -7]]]] </code> </pre> <P> 结果就是这样的张量 <code> dists[i1, .., in, j1, .., jm] </code> 是 <code> a[i1, .., in] - b[j1, .., jm] </code> ，与 <code> n </code> 和 <code> m </code> 是维数 <code> a </code> 和 <code> b </code> 分别。 </p> </DIV>

编辑
1#
回复此人
易米烊光 | 2019-08-31 10-32

<div class =“post-text”itemprop =“text”> <P> 你也可以使用 <code> tf.meshgrid </code> 实现它。 </p> <pre> <code> import tensorflow as tf import numpy as np a = np.array([[1,2,3],[4,5,6],[7,8,9]]) b = np.array([[11,12],[13,14],[15,16]]) a_tf = tf.placeholder(shape=(None,None),dtype=tf.float32) b_tf = tf.placeholder(shape=(None,None),dtype=tf.float32) A,B = tf.meshgrid(a_tf,b_tf) result = tf.transpose(A-B) # two dimension result = tf.reshape(result,shape=(-1,tf.shape(b_tf)[0],tf.shape(b_tf)[1])) # three dimension with tf.Session() as sess: print(sess.run(result, feed_dict={a_tf: a, b_tf: b})) [[[-10. -11.] [-12. -13.] [-14. -15.]] [[ -9. -10.] [-11. -12.] [-13. -14.]] [[ -8. -9.] [-10. -11.] [-12. -13.]] [[ -7. -8.] [ -9. -10.] [-11. -12.]] [[ -6. -7.] [ -8. -9.] [-10. -11.]] [[ -5. -6.] [ -7. -8.] [ -9. -10.]] [[ -4. -5.] [ -6. -7.] [ -8. -9.]] [[ -3. -4.] [ -5. -6.] [ -7. -8.]] [[ -2. -3.] [ -4. -5.] [ -6. -7.]]] </code> </pre> </DIV>

编辑

登录后才能参与评论