Add array_grad._SplitGrad()

Oceania2018 · Oceania2018 · commit 42ba8b6233f0 · 2019-11-24T20:38:47.000-06:00
diff --git a/src/TensorFlowNET.Core/Gradients/array_grad.cs b/src/TensorFlowNET.Core/Gradients/array_grad.cs
@@ -231,6 +231,12 @@ public static Tensor[] _PadGrad(Operation op, Tensor[] grads)
                 return new Tensor[] { x_grad, null };
         }
 
+        [RegisterGradient("Split")]
+        public static Tensor[] _SplitGrad(Operation op, Tensor[] grads)
+        {
+            return new Tensor[] { null, array_ops.concat(list(grads), op.inputs[0]) };
+        }
+
         [RegisterGradient("Squeeze")]
         public static Tensor[] _SqueezeGrad(Operation op, Tensor[] grads)
         {

Original file line number	Diff line number	Diff line change
`@@ -231,6 +231,12 @@ public static Tensor[] _PadGrad(Operation op, Tensor[] grads)`
`231`	`231`	`return new Tensor[] { x_grad, null };`
`232`	`232`	`}`
`233`	`233`
	`234`	`+ [RegisterGradient("Split")]`
	`235`	`+ public static Tensor[] _SplitGrad(Operation op, Tensor[] grads)`
	`236`	`+ {`
	`237`	`+ return new Tensor[] { null, array_ops.concat(list(grads), op.inputs[0]) };`
	`238`	`+ }`
	`239`	`+`
`234`	`240`	`[RegisterGradient("Squeeze")]`
`235`	`241`	`public static Tensor[] _SqueezeGrad(Operation op, Tensor[] grads)`
`236`	`242`	`{`