|
|
@@ -13,6 +13,7 @@ namespace TensorFlowNET.UnitTest.TF_API |
|
|
|
{ |
|
|
|
// A constant vector of size 6 |
|
|
|
Tensor a = tf.constant(new float[] { 1.0f, -0.5f, 3.4f, -2.1f, 0.0f, -6.5f }); |
|
|
|
Tensor b = tf.constant(new float[,] { { 1.0f, -0.5f, 3.4f }, { -2.1f, 0.0f, -6.5f } }); |
|
|
|
|
|
|
|
[TestMethod] |
|
|
|
public void Sin() |
|
|
@@ -31,5 +32,24 @@ namespace TensorFlowNET.UnitTest.TF_API |
|
|
|
var actual = b.ToArray<float>(); |
|
|
|
Assert.IsTrue(Equal(expected, actual)); |
|
|
|
} |
|
|
|
|
|
|
|
[TestMethod] |
|
|
|
public void ReduceSum() |
|
|
|
{ |
|
|
|
var x1 = tf.reduce_sum(b); |
|
|
|
Assert.AreEqual(-4.7f, (float)x1); |
|
|
|
|
|
|
|
var x2 = tf.reduce_sum(b, 0); |
|
|
|
Assert.IsTrue(Enumerable.SequenceEqual(new[] { -1.0999999f, -0.5f, -3.1f }, x2.ToArray<float>())); |
|
|
|
|
|
|
|
var x3 = tf.reduce_sum(b, 1); |
|
|
|
Assert.IsTrue(Enumerable.SequenceEqual(new[] { 3.9f, -8.6f }, x3.ToArray<float>())); |
|
|
|
|
|
|
|
var x4 = tf.reduce_sum(b, 1, keepdims: true); |
|
|
|
Assert.AreEqual((2, 1), x4.TensorShape); |
|
|
|
|
|
|
|
var x5 = tf.reduce_sum(b, (0, 1)); |
|
|
|
Assert.AreEqual(-4.7f, (float)x5); |
|
|
|
} |
|
|
|
} |
|
|
|
} |