| FazBrowse GitHub Viewer | Trending | | Home |
| Tools: [Download Repo ZIP] [Original HTTPS Page] |
1 parent ab7b986 commit 005476c
1 file changed
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
@@ -373,5 +373,13 @@ public static Tensor[] _TransposeGrad(Operation op, Tensor[] grads) | |||
| 373 | 373 | var p = op.inputs[1]; | |
| 374 | 374 | return new Tensor[] { array_ops.transpose(grads[0], array_ops.invert_permutation(p)), null }; | |
| 375 | 375 | } | |
| 376 | + | ||
| 377 | + [RegisterGradient("ReverseV2")] | ||
| 378 | + public static Tensor[] _ReverseV2Grad(Operation op, Tensor[] grads) | ||
| 379 | + { | ||
| 380 | + var grad = grads[0]; | ||
| 381 | + var axis = op.inputs[1]; | ||
| 382 | + return new Tensor[] { array_ops.reverse(grad, axis), null }; | ||
| 383 | + } | ||
| 376 | 384 | } | |
| 377 | 385 | } | |
| Back | FazBrowse Home | New Git URL |
0 commit comments