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.

InitLayerNameTest.cs 1.1 kB

123456789101112131415161718192021222324252627282930313233
  1. using Microsoft.VisualStudio.TestTools.UnitTesting;
  2. using Tensorflow.Keras.Layers;
  3. using static Tensorflow.Binding;
  4. using static Tensorflow.KerasApi;
  5. namespace Tensorflow.Keras.UnitTest
  6. {
  7. [TestClass]
  8. public class InitLayerNameTest
  9. {
  10. [TestMethod]
  11. public void RNNLayerNameTest()
  12. {
  13. var simpleRnnCell = keras.layers.SimpleRNNCell(1);
  14. Assert.AreEqual("simple_rnn_cell", simpleRnnCell.Name);
  15. var simpleRnn = keras.layers.SimpleRNN(2);
  16. Assert.AreEqual("simple_rnn", simpleRnn.Name);
  17. var lstmCell = keras.layers.LSTMCell(2);
  18. Assert.AreEqual("lstm_cell", lstmCell.Name);
  19. var lstm = keras.layers.LSTM(3);
  20. Assert.AreEqual("lstm", lstm.Name);
  21. }
  22. [TestMethod]
  23. public void ConvLayerNameTest()
  24. {
  25. var conv2d = keras.layers.Conv2D(8, activation: "linear");
  26. Assert.AreEqual("conv2d", conv2d.Name);
  27. var conv2dTranspose = keras.layers.Conv2DTranspose(8);
  28. Assert.AreEqual("conv2d_transpose", conv2dTranspose.Name);
  29. }
  30. }
  31. }