fix: add the gradient of the tf.gradient opr

SciSharp · Jul 26, 2023 · 005476c · 005476c
1 parent ab7b986
commit 005476c
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/src/TensorFlowNET.Core/Gradients/array_grad.cs b/src/TensorFlowNET.Core/Gradients/array_grad.cs
@@ -373,5 +373,13 @@ public static Tensor[] _TransposeGrad(Operation op, Tensor[] grads)
             var p = op.inputs[1];
             return new Tensor[] { array_ops.transpose(grads[0], array_ops.invert_permutation(p)), null };
         }
+
+        [RegisterGradient("ReverseV2")]
+        public static Tensor[] _ReverseV2Grad(Operation op, Tensor[] grads)
+        {
+            var grad = grads[0];
+            var axis = op.inputs[1];
+            return new Tensor[] { array_ops.reverse(grad, axis), null };
+        }
     }
 }