finalise_batchnorm2d Subroutine

  subroutine finalise_batchnorm2d(this)
    !! Finalise 2D batch normalisation layer
    implicit none

    ! Arguments
    type(batchnorm2d_layer_type), intent(inout) :: this
    !! Instance of the 2D batch normalisation layer

    if(allocated(this%mean)) deallocate(this%mean)
    if(allocated(this%variance)) deallocate(this%variance)
    if(allocated(this%input_shape)) deallocate(this%input_shape)
    if(allocated(this%output)) deallocate(this%output)

  end subroutine finalise_batchnorm2d