Primitive Sum Reduction

Sums array elements along the specified dimensions.

Usage

prim_reduce_sum(operand, dims, drop = TRUE)

Arguments

operand

(arrayish)
Arrayish value of any data type.

dims

(integer())
Dimensions to reduce over.

drop

(logical(1))
Whether to drop the reduced dimensions from the output shape. If TRUE, the reduced dimensions are removed. If FALSE, the reduced dimensions are set to 1.

Value

arrayish
Has the same data type as the input. When drop = TRUE, the shape is that of operand with dims removed. When drop = FALSE, the shape is that of operand with dims set to 1. It is ambiguous if the input is ambiguous.

Implemented Rules

• stablehlo

• quickr

• reverse

StableHLO

Lowers to stablehlo::hlo_reduce() with stablehlo::hlo_add() as the reducer.

See also

nv_reduce_sum()

Examples

x <- nv_matrix(1:6, nrow = 2)
prim_reduce_sum(x, dims = 1L)
#> AnvlArray
#>   3
#>   7
#>  11
#> [ CPUi32{3} ]