TensorFlow 1: расчет большого расстояния - PullRequest
0 голосов
/ 31 октября 2019

У меня есть два набора n-мерных точек. Эти множества расположены в два тензора, назовите их A и B. Я хочу перебрать точки в A и назначить каждый из них ближайшей точке в B. Однако я хочу сделать это без замены. То есть, как только точка в B назначается точке в A, она недоступна для назначения другой точке в A.

Я хочу реализовать это в TensorFlow. Прямо сейчас, лучшее, что у меня есть, это матрица расстояний (код ниже). Тем не менее, tf.reduce_max( distances , axis=1) не даст мне того, что я хочу, потому что не будет использовать все точки в A и B (при условии, что A имеет больше точек, чем B), как в предыдущем абзаце.

distances = tf.sqrt(tf.reduce_sum((A[:,np.newaxis] - B)**2, axis=2))
...