You can not select more than 25 topics Topics must start with a chinese character,a letter or number, can include dashes ('-') and can be up to 35 characters long.

KerasTests.cs 801 B

6 years ago
6 years ago
6 years ago
6 years ago
6 years ago
6 years ago
123456789101112131415161718192021222324252627
  1. using Tensorflow;
  2. using Keras.Layers;
  3. using NumSharp;
  4. using Microsoft.VisualStudio.TestTools.UnitTesting;
  5. using static Tensorflow.Binding;
  6. namespace TensorFlowNET.UnitTest
  7. {
  8. [TestClass]
  9. public class BaseTests
  10. {
  11. [TestMethod]
  12. public void Dense_Tensor_ShapeTest()
  13. {
  14. var dense_1 = new Dense(1, name: "dense_1", activation: tf.nn.relu());
  15. var input = new Tensor(np.array(new int[] { 3 }));
  16. dense_1.__build__(input.TensorShape);
  17. var outputShape = dense_1.output_shape(input.TensorShape);
  18. var a = (int[])(outputShape.dims);
  19. var b = (int[])(new int[] { 1 });
  20. var _a = np.array(a);
  21. var _b = np.array(b);
  22. Assert.IsTrue(np.array_equal(_a, _b));
  23. }
  24. }
  25. }