Primitive Clamp — nvl_clamp • anvil

Clamps every element of operand to the range [min_val, max_val], i.e. max(min_val, min(operand, max_val)).

Usage

nvl_clamp(min_val, operand, max_val)

Arguments

min_val: (tensorish)
Minimum value. Must be scalar or the same shape as operand.
operand: (tensorish)
Tensorish value of any data type.
max_val: (tensorish)
Maximum value. Must be scalar or the same shape as operand.

Value

tensorish
Has the same data type and shape as operand. It is ambiguous if the input is ambiguous.

Implemented Rules

stablehlo
backward

StableHLO

Lowers to stablehlo::hlo_clamp().

See also

Examples

jit_eval({
  x <- nv_tensor(c(-1, 0.5, 2))
  nvl_clamp(nv_scalar(0), x, nv_scalar(1))
})
#> AnvilTensor
#>  0.0000
#>  0.5000
#>  1.0000
#> [ CPUf32{3} ]