Update edgeconv.py

parent a9baa36b
......@@ -146,4 +146,6 @@ def knn(topk_indices, features):
# topk_indices: (N, P, K)
# features: (N, P, C)
# return: (N, P, K, C)
return tf.gather(features, topk_indices, batch_dims=1)
with tf.name_scope("knn"):
knn_features = tf.gather(features, topk_indices, batch_dims=1)
return knn_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