get_partial_gno_agg_kernels Function

  function get_partial_gno_agg_kernels(this, upstream_grad) result(output)
    !! Gradient of gno_aggregate w.r.t. edge_kernels (right operand)
    !!
    !! d(m_i)/d(kappa_e) = h_j (Kronecker-product structure)
    !! For vectorised kappa: grad_kappa(e) = upstream(:,i) ⊗ h_j
    implicit none

    ! Arguments
    class(array_type), intent(inout) :: this
    !! Forward result node containing saved operands
    type(array_type), intent(in) :: upstream_grad
    !! Upstream gradient tensor
    type(array_type) :: output
    !! Gradient tensor for edge kernels

    call output%allocate(array_shape=shape(this%right_operand%val))
    call this%get_partial_right_val(upstream_grad%val, output%val)

  end function get_partial_gno_agg_kernels