Skip to contents

Element-wise reciprocal square root.

Usage

nvl_rsqrt(operand)

Arguments

operand

(tensorish)
Tensorish value of data type floating-point.

Value

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

Implemented Rules

  • stablehlo

  • backward

StableHLO

Lowers to stablehlo::hlo_rsqrt().

See also

Examples

jit_eval({
  x <- nv_tensor(c(1, 4, 9))
  nvl_rsqrt(x)
})
#> AnvilTensor
#>  1.0000
#>  0.5000
#>  0.3333
#> [ CPUf32{3} ]