Spread output of runInput for both NN and NNGPU, NNGPU does a `toAr…

…ray(this.gpu)` first. This prevents object sharing issues.
sygops · Jan 19, 2018 · 8e84766 · 8e84766
1 parent b057d4c
commit 8e84766
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/src/neural-network-gpu.js b/src/neural-network-gpu.js
@@ -250,7 +250,7 @@ export default class NeuralNetworkGPU extends NeuralNetwork {
     if (this.inputLookup) {
       input = lookup.toArray(this.inputLookup, input);
     }
-    let output = this.runInput(input);
+    let output = [...this.runInput(input).toArray(this.gpu)];
 
     if (this.outputLookup) {
       output = lookup.toHash(this.outputLookup, output);

diff --git a/src/neural-network.js b/src/neural-network.js
@@ -130,7 +130,7 @@ export default class NeuralNetwork {
       input = lookup.toArray(this.inputLookup, input);
     }
 
-    let output = this.runInput(input);
+    let output = [...this.runInput(input)];
 
     if (this.outputLookup) {
       output = lookup.toHash(this.outputLookup, output);