TensorFlow.NET/test/TensorFlowNET.UnitTest/Keras/Layers.Reshaping.Test.cs

using Microsoft.VisualStudio.TestTools.UnitTesting;
using NumSharp;
using static Tensorflow.Binding;
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);
        }

        [TestMethod]
        public void Reshape()
        {
            var inputs = tf.zeros((10, 5, 20));
            var outputs = keras.layers.LeakyReLU().Apply(inputs);
            outputs = keras.layers.Reshape((20, 5)).Apply(outputs);
            Assert.AreEqual((10, 20, 5), outputs.shape);
        }
    }
}