layer_setup Module Function

private module function layer_setup(num_vertex_features, num_time_steps, activation, kernel_initialiser, verbose) result(layer)

Uses

Set up the message passing layer

Arguments

Type IntentOptional Attributes Name
integer, intent(in), dimension(:) :: num_vertex_features

Number of features
integer, intent(in) :: num_time_steps

Number of time steps
class(*), intent(in), optional :: activation

Activation function and kernel initialiser
class(*), intent(in), optional :: kernel_initialiser

Activation function and kernel initialiser
integer, intent(in), optional :: verbose

Verbosity level

Return Value type(kipf_msgpass_layer_type)

Instance of the message passing layer


Source Code

     module function layer_setup( &
          num_vertex_features, num_time_steps, &
          activation, &
          kernel_initialiser, &
          verbose &
     ) result(layer)
       !! Set up the message passing layer
       integer, dimension(:), intent(in) :: num_vertex_features
       !! Number of features
       integer, intent(in) :: num_time_steps
       !! Number of time steps
       class(*), optional, intent(in) :: activation, kernel_initialiser
       !! Activation function and kernel initialiser
       integer, optional, intent(in) :: verbose
       !! Verbosity level
       type(kipf_msgpass_layer_type) :: layer
       !! Instance of the message passing layer
     end function layer_setup