get_partial_exp_val_sum Subroutine

  pure subroutine get_partial_exp_val_sum(this, upstream_grad, output)
    !! Fused partial+sum for exp: output(:,1) = sum_s(upstream * exp(x))
    implicit none
    class(array_type), intent(in) :: this
    real(real32), dimension(:,:), intent(in) :: upstream_grad
    real(real32), dimension(:), intent(out) :: output

    integer :: s, n_samples

    n_samples = size(upstream_grad, 2)
    if(.not.this%is_sample_dependent)then
       output(:) = upstream_grad(:,1)
       do s = 2, n_samples
          output(:) = output(:) + upstream_grad(:,s)
       end do
       output(:) = output(:) * this%val(:,1)
    else
       output(:) = upstream_grad(:,1) * this%val(:,1)
       do s = 2, n_samples
          output(:) = output(:) + upstream_grad(:,s) * this%val(:,s)
       end do
    end if
  end subroutine get_partial_exp_val_sum