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.4 kB

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