print_to_unit_flatten Subroutine

  subroutine print_to_unit_flatten(this, unit)
    !! Print flatten layer to unit
    implicit none

    ! Arguments
    class(flatten_layer_type), intent(in) :: this
    !! Instance of the flatten layer
    integer, intent(in) :: unit
    !! File unit

    ! Local variables
    character(100) :: fmt
    !! Format string


    ! Write initial parameters
    !---------------------------------------------------------------------------
    write(unit,'(3X,"INPUT_RANK = ",I0)') this%input_rank
    write(fmt,'("(3X,""INPUT_SHAPE ="",",I0,"(1X,I0))")') size(this%input_shape)
    write(unit,fmt) this%input_shape

  end subroutine print_to_unit_flatten