Browse Source

Add array_grad._SplitGrad()

tags/v0.13
Oceania2018 6 years ago
parent
commit
42ba8b6233
1 changed files with 6 additions and 0 deletions
  1. +6
    -0
      src/TensorFlowNET.Core/Gradients/array_grad.cs

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

@@ -231,6 +231,12 @@ namespace Tensorflow.Gradients
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)
{


Loading…
Cancel
Save