[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