|
|
@@ -0,0 +1,33 @@ |
|
|
|
using Microsoft.VisualStudio.TestTools.UnitTesting; |
|
|
|
using Tensorflow.Keras.Layers; |
|
|
|
using static Tensorflow.Binding; |
|
|
|
using static Tensorflow.KerasApi; |
|
|
|
|
|
|
|
namespace Tensorflow.Keras.UnitTest |
|
|
|
{ |
|
|
|
[TestClass] |
|
|
|
public class InitLayerNameTest |
|
|
|
{ |
|
|
|
[TestMethod] |
|
|
|
public void RNNLayerNameTest() |
|
|
|
{ |
|
|
|
var simpleRnnCell = keras.layers.SimpleRNNCell(1); |
|
|
|
Assert.AreEqual("simple_rnn_cell", simpleRnnCell.Name); |
|
|
|
var simpleRnn = keras.layers.SimpleRNN(2); |
|
|
|
Assert.AreEqual("simple_rnn", simpleRnn.Name); |
|
|
|
var lstmCell = keras.layers.LSTMCell(2); |
|
|
|
Assert.AreEqual("lstm_cell", lstmCell.Name); |
|
|
|
var lstm = keras.layers.LSTM(3); |
|
|
|
Assert.AreEqual("lstm", lstm.Name); |
|
|
|
} |
|
|
|
|
|
|
|
[TestMethod] |
|
|
|
public void ConvLayerNameTest() |
|
|
|
{ |
|
|
|
var conv2d = keras.layers.Conv2D(8, activation: "linear"); |
|
|
|
Assert.AreEqual("conv2d", conv2d.Name); |
|
|
|
var conv2dTranspose = keras.layers.Conv2DTranspose(8); |
|
|
|
Assert.AreEqual("conv2d_transpose", conv2dTranspose.Name); |
|
|
|
} |
|
|
|
} |
|
|
|
} |