Clamp

Element-wise clamp: max(min_val, min(operand, max_val)). Converts min_val and max_val to the data type of operand.

Usage

nv_clamp(min_val, operand, max_val)

Arguments

min_val, max_val

(tensorish)
Minimum and maximum values (scalar or same shape as operand).

operand

(tensorish)
Operand.

Value

tensorish
Has the same shape and data type as the input.

Details

The underlying stableHLO function already broadcasts scalars, so no need to broadcast manually.

See also

nvl_clamp() for the underlying primitive.

Examples

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