get_partial_concat_right_val Subroutine

pure subroutine get_partial_concat_right_val(this, upstream_grad, output)

Arguments

Type IntentOptional Attributes Name
class(array_type), intent(in) :: this
real(kind=real32), intent(in), dimension(:,:) :: upstream_grad
real(kind=real32), intent(out), dimension(:,:) :: output

Source Code

  pure subroutine get_partial_concat_right_val(this, upstream_grad, output)
    implicit none
    class(array_type), intent(in) :: this
    real(real32), dimension(:,:), intent(in) :: upstream_grad
    real(real32), dimension(:,:), intent(out) :: output

    if(this%indices(1).eq.1)then
       output = upstream_grad( &
            size(this%left_operand%val,1)+1:size(upstream_grad,1), : &
       )
    else
       output = upstream_grad(:, &
            size(this%left_operand%val,2)+1:size(upstream_grad,2) &
       )
    end if

  end subroutine get_partial_concat_right_val