Browse Source

fix test

pull/802/head
MPnoy 4 years ago
parent
commit
03b28cd945
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      test/TensorFlowNET.UnitTest/ManagedAPI/GradientTest.cs

+ 1
- 1
test/TensorFlowNET.UnitTest/ManagedAPI/GradientTest.cs View File

@@ -68,8 +68,8 @@ namespace TensorFlowNET.UnitTest.ManagedAPI
var X = tf.zeros(new TensorShape(10)); var X = tf.zeros(new TensorShape(10));
var W = tf.Variable(-0.06f, name: "weight"); var W = tf.Variable(-0.06f, name: "weight");
var b = tf.Variable(-0.73f, name: "bias"); var b = tf.Variable(-0.73f, name: "bias");
var test = tf.concat(new Tensor[] { W, b }, 0);
using var g = tf.GradientTape(); using var g = tf.GradientTape();
var test = tf.concat(new Tensor[] { W, b }, 0);
var pred = test[0] * X + test[1]; var pred = test[0] * X + test[1];
var gradients = g.gradient(pred, (W, b)); var gradients = g.gradient(pred, (W, b));
Assert.AreEqual((float)gradients.Item1, 0); Assert.AreEqual((float)gradients.Item1, 0);


Loading…
Cancel
Save