forward_msgpass Module Subroutine

module subroutine forward_msgpass(this, input)

Forward propagation for the layer

Arguments

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

Instance of the layer type
class(array_type), intent(in), dimension(:,:) :: input

Input data (i.e. vertex and edge features)

Source Code

  module subroutine forward_msgpass(this, input)
    !! Forward propagation for the layer
    implicit none

    ! Arguments
    class(msgpass_layer_type), intent(inout) :: this
    !! Instance of the layer type
    class(array_type), dimension(:,:), intent(in) :: input
    !! Input data (i.e. vertex and edge features)


    call this%update_message(input)
    call this%update_readout()

  end subroutine forward_msgpass