All Reduction — nv_reduce_all • anvil

Performs logical AND along the specified dimensions. Returns TRUE only if all elements are TRUE.

Usage

nv_reduce_all(operand, dims, drop = TRUE)

Arguments

operand: (tensorish)
Operand.
dims: (integer())
Dimensions to reduce.
drop: (logical(1))
Whether to drop reduced dimensions.

Value

tensorish
Boolean tensor. When drop = TRUE, the reduced dimensions are removed. When drop = FALSE, the reduced dimensions are set to 1.

See also

nvl_reduce_all() for the underlying primitive.

Examples

jit_eval({
  x <- nv_tensor(matrix(c(TRUE, FALSE, TRUE, TRUE), nrow = 2))
  nv_reduce_all(x, dims = 1L)
})
#> AnvilTensor
#>  0
#>  1
#> [ CPUi1{2} ]