wassname/Flex-Convolution
1
0
Fork 0
mirror of https://github.com/wassname/Flex-Convolution.git synced 2026-09-09 11:14:10 +08:00
Code Issues Packages Projects Releases Wiki Activity

typo: remove hardcoding of K in _knn_bruteforce

Browse Source 
I believe that you meant for this to be self.K not 4, please correct me if I am wrong.
This commit is contained in:
Mike Clark
2018-11-18 15:49:28 +08:00
committed by GitHub
parent 4ec3734736
commit 497bebd041
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
layers.py
+1 -1
View File
@@ -81,7 +81,7 @@ class KnnBruteforce(Layer):
if self.data_format == 'expanded':
positions = _remove_dim(positions, 2)
NN, _, _ = _knn_bruteforce(positions, K=4)
NN, _, _ = _knn_bruteforce(positions, K=self.K)
NN = tf.transpose(NN, [0, 2, 1])
if self.data_format == 'expanded':