Yes, a lot of tests are missed for many layers. It will be great to add them to the test suite

Hey @zaleslaw, would it be nice if I put Merge tests in one file as the ReshapeTest looks tidy but what are your thoughts?

Yes, put it into one file

Hey @zaleslaw , have a look at this test case, this is from tf official doc yet it fails, AFAIK the assertLayerOutputIsCorrect checks the content of the output as well and Average() should produce the same effect as given in the doc. Yet the test fails and when i tried to debug it it just returns the [x1,x2] as result.

@Test
    fun average(){
        val x1 = Array(2){FloatArray(2){1f} }
        val x2 = Array(2){FloatArray(2){0f} }
        val input = arrayOf(x1,x2)
        val expected = Array(2){FloatArray(2){0.5f} }
        assertLayerOutputIsCorrect(Average(), input,expected)
    }

Thanks for the note @therealansh ! Probably the Average layer could be implemented with a bug (low probability), or you could find something else (high probability) that influences on the output

Hey @zaleslaw, I guess the bug is in LayerTest.kt. Since, Merge layer abstraction has a overloading of forward function, i guess we need overload this as well getLayerOutputOp() as it takes Array<*> which returns a Operand rather than List<Operand<Float>> I guess this can be the only reason for it. Otherwise the Average layer is implemented correctly.