|
|
@@ -0,0 +1,30 @@ |
|
|
|
using Microsoft.VisualStudio.TestTools.UnitTesting; |
|
|
|
using NumSharp; |
|
|
|
using Tensorflow; |
|
|
|
using static Tensorflow.KerasApi; |
|
|
|
|
|
|
|
namespace TensorFlowNET.UnitTest.Keras |
|
|
|
{ |
|
|
|
[TestClass] |
|
|
|
public class LayersReshapingTest : EagerModeTestBase |
|
|
|
{ |
|
|
|
[TestMethod] |
|
|
|
public void ZeroPadding2D() |
|
|
|
{ |
|
|
|
var input_shape = new[] { 1, 1, 2, 2 }; |
|
|
|
var x = np.arange(np.prod(input_shape)).reshape(input_shape); |
|
|
|
var zero_padding_2d = keras.layers.ZeroPadding2D(new[,] { { 1, 0 }, { 1, 0 } }); |
|
|
|
var y = zero_padding_2d.Apply(x); |
|
|
|
Assert.AreEqual((1, 2, 3, 2), y.shape); |
|
|
|
} |
|
|
|
|
|
|
|
[TestMethod] |
|
|
|
public void UpSampling2D() |
|
|
|
{ |
|
|
|
var input_shape = new[] { 2, 2, 1, 3 }; |
|
|
|
var x = np.arange(np.prod(input_shape)).reshape(input_shape); |
|
|
|
var y = keras.layers.UpSampling2D(size: (1, 2)).Apply(x); |
|
|
|
Assert.AreEqual((2, 2, 2, 3), y.shape); |
|
|
|
} |
|
|
|
} |
|
|
|
} |