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.

ModelBuildTest.cs 1.5 kB

123456789101112131415161718192021222324252627282930313233343536373839404142
  1. using Microsoft.VisualStudio.TestTools.UnitTesting;
  2. using System;
  3. using System.Collections.Generic;
  4. using System.Linq;
  5. using System.Text;
  6. using System.Threading.Tasks;
  7. using static Tensorflow.Binding;
  8. namespace TensorflowNET.Keras
  9. {
  10. [TestClass]
  11. public class ModelBuildTest
  12. {
  13. [TestMethod]
  14. public void DenseBuild()
  15. {
  16. // two dimensions input with unknown batchsize
  17. var input = tf.keras.layers.Input((17, 60));
  18. var dense = tf.keras.layers.Dense(64);
  19. var output = dense.Apply(input);
  20. var model = tf.keras.Model(input, output);
  21. // one dimensions input with unknown batchsize
  22. var input_2 = tf.keras.layers.Input((60));
  23. var dense_2 = tf.keras.layers.Dense(64);
  24. var output_2 = dense.Apply(input_2);
  25. var model_2 = tf.keras.Model(input_2, output_2);
  26. // two dimensions input with specified batchsize
  27. var input_3 = tf.keras.layers.Input((17, 60), 8);
  28. var dense_3 = tf.keras.layers.Dense(64);
  29. var output_3 = dense.Apply(input_3);
  30. var model_3 = tf.keras.Model(input_3, output_3);
  31. // one dimensions input with specified batchsize
  32. var input_4 = tf.keras.layers.Input((60), 8);
  33. var dense_4 = tf.keras.layers.Dense(64);
  34. var output_4 = dense.Apply(input_4);
  35. var model_4 = tf.keras.Model(input_4, output_4);
  36. }
  37. }
  38. }