Adding TNNetSplitChannels.Create(ChannelStart, ChannelLen: integer)

joaopauloschuler · Dec 3, 2019 · 9600a72 · 9600a72
1 parent 1ef4b10
commit 9600a72
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/neural/neuralnetwork.pas b/neural/neuralnetwork.pas
@@ -664,6 +664,7 @@   TNNetSplitChannels = class(TNNetLayer)
     FChannels: TNeuralIntegerArray;
     procedure SetPrevLayer(pPrevLayer: TNNetLayer); override;
   public
+    constructor Create(ChannelStart, ChannelLen: integer); overload;
     constructor Create(pChannels: array of integer); overload;
     destructor Destroy(); override;
 
@@ -3833,6 +3834,17 @@ procedure TNNetSplitChannels.SetPrevLayer(pPrevLayer: TNNetLayer);
   FActivationFnDerivative := pPrevLayer.ActivationFnDerivative;
 end;
 
+constructor TNNetSplitChannels.Create(ChannelStart, ChannelLen: integer);
+var
+  pChannels: array of integer;
+  ChannelCnt: integer;
+begin
+  SetLength(pChannels, ChannelLen);
+  for ChannelCnt := 0 to ChannelLen - 1 do
+    pChannels[ChannelCnt] := ChannelStart + ChannelCnt;
+  Create(pChannels);
+end;
+
 constructor TNNetSplitChannels.Create(pChannels: array of integer);
 var
   I: integer;