Primitive Square Root

Element-wise square root.

Usage

prim_sqrt(operand)

Arguments

operand

(arrayish)
Arrayish value of data type floating-point.

Value

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

Implemented Rules

• stablehlo

• quickr

• reverse

StableHLO

Lowers to stablehlo::hlo_sqrt().

See also

nv_sqrt(), sqrt()

Examples

x <- nv_array(c(1, 4, 9))
prim_sqrt(x)
#> AnvlArray
#>  1
#>  2
#>  3
#> [ CPUf32{3} ]