Primitive Square Root

Element-wise square root.

Usage

nvl_sqrt(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_sqrt().

See also

nv_sqrt(), sqrt()

Examples

jit_eval({
  x <- nv_tensor(c(1, 4, 9))
  nvl_sqrt(x)
})
#> AnvilTensor
#>  1
#>  2
#>  3
#> [ CPUf32{3} ]