shown input shape corrected

saliherdemk · Nov 6, 2024 · 183d1d8 · 183d1d8
1 parent 81a9ea2
commit 183d1d8
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/js/Draw/MLP/Playable.js b/js/Draw/MLP/Playable.js
@@ -315,7 +315,7 @@ class Playable extends Draggable {
     const calcComp = this.calculationComponent;
     const output = this.getOutput();
 
-    calcComp?.setInputData(inputData.slice(0, 5).map((row) => row.slice(0, 4)));
+    calcComp?.setInputData(inputData);
 
     const outputData = output instanceof OutputLayer ? output.getData() : null;
     const mlp_output = await origin.forward(inputData);

diff --git a/js/Draw/Minors/Viewers/CalculationViewer.js b/js/Draw/Minors/Viewers/CalculationViewer.js
@@ -20,6 +20,7 @@ class CalculationViewer extends Viewer {
 
   setInputData(data) {
     const shape = [data.length, data[0].length];
+    data = data.slice(0, 5).map((row) => row.slice(0, 4));
     this.data = [this.formatMatrix(data, shape, 0, 0)];
   }