From 5b68125c9cad912775d4b184b10ecad7aaa2e152 Mon Sep 17 00:00:00 2001 From: lucas_miranda <lucasmiranda42@gmail.com> Date: Tue, 18 May 2021 19:40:48 +0200 Subject: [PATCH] Replaced for loop with vectorised mapping on ClusterOverlap regularization layer --- deepof/model_utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/deepof/model_utils.py b/deepof/model_utils.py index 37356da8..dbe1d3b4 100644 --- a/deepof/model_utils.py +++ b/deepof/model_utils.py @@ -39,7 +39,7 @@ def compute_shannon_entropy(tensor): @tf.function def get_k_nearest_neighbors(tensor, k, index): """Retrieve indices of the k nearest neighbors in tensor to the vector with the specified index""" - query = tf.slice(tensor, index, 1) + query = tensor[index] distances = tf.norm(tensor - query, axis=1) max_distance = tf.sort(distances)[k] neighbourhood_mask = distances < max_distance -- GitLab