Skip to contents

Sum Reduction

Source: R/api.R
nv_reduce_sum.Rd

Sums tensor elements along the specified dimensions.

Usage

nv_reduce_sum(operand, dims, drop = TRUE)

Arguments

operand

(tensorish)
Operand.

dims

(integer())
Dimensions to reduce.

drop

(logical(1))
Whether to drop reduced dimensions.

Value

tensorish
Has the same data type as the input. When drop = TRUE, the reduced dimensions are removed. When drop = FALSE, the reduced dimensions are set to 1.

See also

nvl_reduce_sum() for the underlying primitive.

Examples

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