美文网首页
如何在tensorflow计算图定义批量计算线距离的op

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

作者: 点指兵兵_ | 来源:发表于2019-03-20 21:37 被阅读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 就可以啦
大家有什么好的想法也可以交流下。

相关文章

网友评论

      本文标题:如何在tensorflow计算图定义批量计算线距离的op

      本文链接:https://www.haomeiwen.com/subject/pocxvqtx.html