假设某几根线是label,要和若干线计算距离,当然也可以两两计算,思路是差不多的。
tf.reshape(tf.tile(label,[1,line_num]),[line_num*lable_num,4])
tf.tile(line,[label_num,1])
两个矩阵的复制的方向是不一样的,大家注意了
这样两个矩阵就是相同的shape了
然后再用tf.squared_difference 就可以啦
大家有什么好的想法也可以交流下。
假设某几根线是label,要和若干线计算距离,当然也可以两两计算,思路是差不多的。
tf.reshape(tf.tile(label,[1,line_num]),[line_num*lable_num,4])
tf.tile(line,[label_num,1])
两个矩阵的复制的方向是不一样的,大家注意了
这样两个矩阵就是相同的shape了
然后再用tf.squared_difference 就可以啦
大家有什么好的想法也可以交流下。
本文标题:如何在tensorflow计算图定义批量计算线距离的op
本文链接:https://www.haomeiwen.com/subject/pocxvqtx.html
网友评论