diff --git a/‎src/TensorFlowNET.Keras/Engine/Layer.Layers.cs‎
Lines changed: 5 additions & 1 deletion b/‎src/TensorFlowNET.Keras/Engine/Layer.Layers.cs‎
Lines changed: 5 additions & 1 deletion
diff --git a/‎src/TensorFlowNET.Keras/Layers/Reshaping/Reshape.cs‎
Lines changed: 2 additions & 2 deletions b/‎src/TensorFlowNET.Keras/Layers/Reshaping/Reshape.cs‎
Lines changed: 2 additions & 2 deletions
@@ -1,4 +1,5 @@
-using System.Collections.Generic;
+using System;
+using System.Collections.Generic;
 
 namespace Tensorflow.Keras.Engine
 {
@@ -11,5 +12,8 @@ protected void StackLayers(params ILayer[] layers)
         {
             _layers.AddRange(layers);
         }
+
+        public virtual TensorShape ComputeOutputShape(TensorShape input_shape)
+            => throw new NotImplementedException("");
     }
 }
@@ -26,11 +26,11 @@ protected override Tensors Call(Tensors inputs, Tensor state = null, bool is_tra
 
             var result = array_ops.reshape(inputs, shape.ToArray());
             if (!tf.Context.executing_eagerly())
-                result.set_shape(compute_output_shape(inputs.shape));
+                result.set_shape(ComputeOutputShape(inputs.shape));
             return result;
         }
 
-        TensorShape compute_output_shape(TensorShape input_shape)
+        public override TensorShape ComputeOutputShape(TensorShape input_shape)
         {
             if (input_shape.dims[0] == -1)
             {
Original file line number	Diff line number	Diff line change
@@ -1,4 +1,5 @@
1		- using System.Collections.Generic;
	1	+ using System;
	2	+ using System.Collections.Generic;
2	3
3	4	namespace Tensorflow.Keras.Engine
4	5	{
@@ -11,5 +12,8 @@ protected void StackLayers(params ILayer[] layers)
11	12	{
12	13	_layers.AddRange(layers);
13	14	}
	15	+
	16	+ public virtual TensorShape ComputeOutputShape(TensorShape input_shape)
	17	+ => throw new NotImplementedException("");
14	18	}
15	19	}
Original file line number	Diff line number	Diff line change
@@ -26,11 +26,11 @@ protected override Tensors Call(Tensors inputs, Tensor state = null, bool is_tra
26	26
27	27	var result = array_ops.reshape(inputs, shape.ToArray());
28	28	if (!tf.Context.executing_eagerly())
29		- result.set_shape(compute_output_shape(inputs.shape));
	29	+ result.set_shape(ComputeOutputShape(inputs.shape));
30	30	return result;
31	31	}
32	32
33		- TensorShape compute_output_shape(TensorShape input_shape)
	33	+ public override TensorShape ComputeOutputShape(TensorShape input_shape)
34	34	{
35	35	if (input_shape.dims[0] == -1)
36	36	{