Add a few tests for a convolutional network; no training yet

Author: milancurcic

CMakeLists.txt

@@ -100,7 +100,7 @@ string(REGEX REPLACE "^ | $" "" LIBS "${LIBS}")
 
 # tests
 enable_testing()
-foreach(execid input1d_layer input3d_layer dense_layer conv2d_layer dense_network)
+foreach(execid input1d_layer input3d_layer dense_layer conv2d_layer dense_network conv2d_network)
   add_executable(test_${execid} test/test_${execid}.f90)
   target_link_libraries(test_${execid} neural ${LIBS})
   add_test(test_${execid} bin/test_${execid})

test/test_conv2d_network.f90

@@ -0,0 +1,42 @@
+program test_conv2d_network
+
+  use iso_fortran_env, only: stderr => error_unit
+  use nf, only: conv2d, input, network
+
+  implicit none
+
+  type(network) :: net
+  real, allocatable :: sample_input(:,:,:), output(:,:,:)
+  logical :: ok = .true.
+
+  ! 3-layer convolutional network
+  net = network([ &
+    input([3, 32, 32]), &
+    conv2d(window_size=3, filters=16), &
+    conv2d(window_size=3, filters=32) &
+  ])
+
+  if (.not. size(net % layers) == 3) then
+    write(stderr, '(a)') 'conv2d network should have 3 layers.. failed'
+    ok = .false.
+  end if
+
+  allocate(sample_input(3, 32, 32))
+  sample_input = 0
+
+  call net % forward(sample_input)
+  call net % layers(3) % get_output(output)
+
+  if (.not. all(shape(output) == [32, 28, 28])) then
+    write(stderr, '(a)') 'conv2d network output should have correct shape.. failed'
+    ok = .false.
+  end if
+
+  if (ok) then
+    print '(a)', 'test_dense_network: All tests passed.'
+  else
+    write(stderr, '(a)') 'test_dense_network: One or more tests failed.'
+    stop 1
+  end if
+
+end program test_conv2d_network