Skip to contents

Primitive Sum Reduction

Source: R/primitives.R
nvl_reduce_sum.Rd

Sums tensor elements along the specified dimensions.

Usage

nvl_reduce_sum(operand, dims, drop = TRUE)

Arguments

operand

(tensorish)
Tensorish 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

tensorish
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

  • backward

StableHLO

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

See also

nv_reduce_sum()

Examples

jit_eval({
  x <- nv_tensor(matrix(1:6, nrow = 2))
  nvl_reduce_sum(x, dims = 1L)
})
#> AnvilTensor
#>   3
#>   7
#>  11
#> [ CPUi32{3} ]