get_partial_val_sum Interface

interface
private module subroutine get_partial_val_sum(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

Description

Sum-reduced partial derivative: output = sum(partial(upstream_grad), dim=2) Avoids allocating large (n_elem, num_samples) intermediate array