diff --git a/test/TensorFlowNET.Keras.UnitTest/Layers/LayersTest.cs b/test/TensorFlowNET.Keras.UnitTest/Layers/LayersTest.cs
index 63e959f5..ba54a0f4 100644
--- a/test/TensorFlowNET.Keras.UnitTest/Layers/LayersTest.cs
+++ b/test/TensorFlowNET.Keras.UnitTest/Layers/LayersTest.cs
@@ -40,7 +40,7 @@ namespace TensorFlowNET.Keras.UnitTest
///
/// Custom layer test, used in Dueling DQN
///
- [TestMethod]
+ [TestMethod, Ignore]
public void TensorFlowOpLayer()
{
var layers = keras.layers;
@@ -58,6 +58,9 @@ namespace TensorFlowNET.Keras.UnitTest
metrics: new[] { "acc" });
model.summary();
Assert.AreEqual(model.Layers.Count, 8);
+ var result = model.predict(tf.constant(np.arange(24).astype(np.float32)[np.newaxis, Slice.All]));
+ Assert.AreEqual(result.shape, new TensorShape(1, 24));
+ model.fit(np.arange(24).astype(np.float32)[np.newaxis, Slice.All], np.arange(24).astype(np.float32)[np.newaxis, Slice.All], verbose: 0);
}
///