TensorFlow.NET/test/TensorFlowNET.UnitTest/KerasTests.cs at master · killbacon/TensorFlow.NET · GitHub

27 lines (25 loc) · 801 Bytes

using Tensorflow;

using Keras.Layers;

using NumSharp;

using Microsoft.VisualStudio.TestTools.UnitTesting;

using static Tensorflow.Binding;

namespace TensorFlowNET.UnitTest

{

[TestClass]

public class BaseTests

{

[TestMethod]

public void Dense_Tensor_ShapeTest()

{

var dense_1 = new Dense(1, name: "dense_1", activation: tf.nn.relu());

var input = new Tensor(np.array(new int[] { 3 }));

dense_1.__build__(input.TensorShape);

var outputShape = dense_1.output_shape(input.TensorShape);

var a = (int[])(outputShape.dims);

var b = (int[])(new int[] { 1 });

var _a = np.array(a);

var _b = np.array(b);

Assert.IsTrue(np.array_equal(_a, _b));

}

Expand file tree