Merge pull request #333 from chelu/different-size-layer

cogmission · cogmission · commit b080249e1db7 · 2015-10-20T20:26:34.000-05:00
Fix computing flat index in SpatialPooler#getNeighborsND
diff --git a/src/main/java/org/numenta/nupic/algorithms/SpatialPooler.java b/src/main/java/org/numenta/nupic/algorithms/SpatialPooler.java
@@ -784,7 +784,7 @@ public TIntArrayList getNeighborsND(Connections c, int columnIndex, SparseMatrix
         TIntArrayList neighbors = new TIntArrayList(neighborList.size());
         int size = neighborList.size();
         for(int i = 0;i < size;i++) {
-            int flatIndex = c.getInputMatrix().computeIndex(neighborList.get(i), false);
+            int flatIndex = topology.computeIndex(neighborList.get(i), false);
             if(flatIndex == columnIndex) continue;
             neighbors.add(flatIndex);
         }

Original file line number	Diff line number	Diff line change
`@@ -784,7 +784,7 @@ public TIntArrayList getNeighborsND(Connections c, int columnIndex, SparseMatrix`
`784`	`784`	`TIntArrayList neighbors = new TIntArrayList(neighborList.size());`
`785`	`785`	`int size = neighborList.size();`
`786`	`786`	`for(int i = 0;i < size;i++) {`
`787`		`- int flatIndex = c.getInputMatrix().computeIndex(neighborList.get(i), false);`
	`787`	`+ int flatIndex = topology.computeIndex(neighborList.get(i), false);`
`788`	`788`	`if(flatIndex == columnIndex) continue;`
`789`	`789`	`neighbors.add(flatIndex);`
`790`	`790`	`}`