Aufgrund einer Wartung wird GitLab am 29.10. zwischen 9:00 und 10:00 Uhr kurzzeitig nicht zur Verfügung stehen. / Due to maintenance, GitLab will be temporarily unavailable on 29.10. between 9:00 and 10:00 am.

[edgeconf] knn: adapted naming

parent 2d34266f
......@@ -147,5 +147,5 @@ def knn(topk_indices, features):
# features: (N, P, C)
# return: (N, P, K, C)
with tf.name_scope("knn"):
knn_features = tf.gather(features, topk_indices, batch_dims=1)
return knn_features
topk_features = tf.gather(features, topk_indices, batch_dims=1)
return topk_features
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment