如何在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 就可以啦
大家有什么好的想法也可以交流下。