Browse Source

add GradientOperatorMulTest 。

tags/v0.30
久永 Haiping 4 years ago
parent
commit
05561ead44
1 changed files with 12 additions and 0 deletions
  1. +12
    -0
      test/TensorFlowNET.UnitTest/ManagedAPI/GradientTest.cs

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

@@ -35,5 +35,17 @@ namespace TensorFlowNET.UnitTest.ManagedAPI
var y_grad = tape.gradient(y, x);
Assert.AreEqual(9.0, (double)y);
}

[TestMethod]
public void GradientOperatorMulTest()
{
var x = tf.constant(0f);
var w = tf.Variable(new float[] { 1, 1 });
using var gt = tf.GradientTape();
var y = x * w;
var gr = gt.gradient(y, w);
Assert.AreNotEqual(null, gr);
}

}
}

Loading…
Cancel
Save