Fix output array size in Trainer class initialization

Author: dlidstrom

CSharp/Neural.cs

@@ -86,7 +86,7 @@ public record Trainer(
     public static Trainer Create(Network network, int hiddenCount, int outputCount)
     {
         double[] hidden = new double[hiddenCount];
-        double[] output = new double[hiddenCount];
+        double[] output = new double[outputCount];
         double[] gradHidden = new double[hiddenCount];
         double[] gradOutput = new double[outputCount];
         return new Trainer(network, hidden, output, gradHidden, gradOutput);

Cpp/neural.cpp

@@ -61,7 +61,7 @@ Vector Network::Predict(const Vector& input, Vector& hidden, Vector& output) con
 
 Trainer Trainer::Create(Neural::Network&& network, size_t hiddenCount, size_t outputCount) {
     Vector hidden = Vector(hiddenCount);
-    Vector output = Vector(hiddenCount);
+    Vector output = Vector(outputCount);
     Vector gradHidden = Vector(hiddenCount);
     Vector gradOutput = Vector(outputCount);
     return Trainer {