Testing Grouped Dot Product #60

joaopauloschuler · Jun 23, 2021 · 3e3e13e · 3e3e13e
1 parent 652e2f2
commit 3e3e13e
Showing 1 changed file with 18 additions and 0 deletions.
diff --git a/neural/neuralvolume.pas b/neural/neuralvolume.pas
@@ -1373,6 +1373,7 @@ procedure TestTNNetVolume();
   B.Resize(A);
   A.FillForDebug();
   B.Copy(A);
+
   Write('Pearson Correlation (A,A):', A.PearsonCorrelation(B) );
   WriteLnPassIfZero(A.PearsonCorrelation(B)-1);
   B.Mul(-1);
@@ -1381,6 +1382,23 @@ procedure TestTNNetVolume();
   B.Randomize();
   Write('Pearson Correlation (A,Random):', A.PearsonCorrelation(B) );
   WriteLnPassIfZero(A.PearsonCorrelation(B), 0.1);
+
+  // Testing Grouped Dot Product
+  // 2 vectors of 2 elements
+  A.Resize(1,2,2); // 0 1 and 2 3
+  // 2 vectors of 4 elements
+  B.Resize(1,2,4); // 0 1 2 3 and 4 5 6 7
+  // 1 resulting vector with 4 elements
+  R.Resize(1,1,4);
+
+  A.FillForDebug();
+  B.FillForDebug();
+  A.Mul(100);
+  B.Mul(100);
+  Write('Grouped dot product result:');
+  R.GroupedDotProductsTiled({Groups=}2, {NumAs=}2, {NumBs=}2,{VectorSize=}2, A, B, {TileSizeA=}1, {TileSizeB=}1);
+  //R.Print();
+  WriteLnPassIfZero(R.GetSum() -1 -5 -13 -33);
   R.Free;
   B.Free;
   A.Free;