set_hyperparams_reshape Subroutine

private subroutine set_hyperparams_reshape(this, output_shape, verbose)

Set hyperparameters for reshape layer

Type Bound

reshape_layer_type

Arguments

Type IntentOptional Attributes Name
class(reshape_layer_type), intent(inout) :: this

Instance of the reshape layer

integer, intent(in), dimension(:) :: output_shape

Output rank

integer, intent(in), optional :: verbose

Verbosity level


Source Code

  subroutine set_hyperparams_reshape(this, output_shape, verbose)
    !! Set hyperparameters for reshape layer
    implicit none

    ! Arguments
    class(reshape_layer_type), intent(inout) :: this
    !! Instance of the reshape layer
    integer, dimension(:), intent(in) :: output_shape
    !! Output rank
    integer, optional, intent(in) :: verbose
    !! Verbosity level

    ! Local variables
    integer :: verbose_ = 0
    !! Verbosity level

    if(present(verbose)) verbose_ = verbose

    this%type = "rshp"
    this%name = "reshape"
    this%input_rank = 0
    this%output_shape = output_shape
    this%output_rank = size(output_shape)

    if(verbose_ .gt. 0) write(*,'("  Setting up reshape layer")')

  end subroutine set_hyperparams_reshape