Browse Source

Merge branch 'SciSharp:master' into master

pull/1158/head^2
Dogvane Huang GitHub 2 years ago
parent
commit
32e540f65f
No known key found for this signature in database GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 8 additions and 0 deletions
  1. +8
    -0
      src/TensorFlowNET.Core/Gradients/array_grad.cs

+ 8
- 0
src/TensorFlowNET.Core/Gradients/array_grad.cs View File

@@ -373,5 +373,13 @@ namespace Tensorflow.Gradients
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 };
}
}
}

Loading…
Cancel
Save