如何在tensorflow计算图定义批量计算线距离的op

2019-03-20  本文已影响0人  点指兵兵_

假设某几根线是label,要和若干线计算距离,当然也可以两两计算,思路是差不多的。

tf.reshape(tf.tile(label,[1,line_num]),[line_num*lable_num,4])
tf.tile(line,[label_num,1])

两个矩阵的复制的方向是不一样的,大家注意了

这样两个矩阵就是相同的shape了
然后再用tf.squared_difference 就可以啦
大家有什么好的想法也可以交流下。

上一篇 下一篇

猜你喜欢

热点阅读