rsqrt

Contents

rsqrt#

class saiunit.lax.rsqrt(x, **kwargs)#

Elementwise reciprocal square root: \(1 \over \sqrt{x}\).

Parameters:

x (Array | ndarray | bool | number | bool | int | float | complex | saiunit.Quantity) – Input array.

Returns:

result – The reciprocal square root. If x has unit u, the result has unit u ** -0.5.

Return type:

saiunit.Quantity | Array

Examples

>>> import saiunit as u
>>> import saiunit.lax as sulax
>>> import jax.numpy as jnp
>>> q = jnp.array([4.0, 9.0, 16.0]) * (u.meter ** 2)
>>> result = sulax.rsqrt(q)
>>> result.mantissa
Array([0.5       , 0.33333334, 0.25      ], dtype=float32)
Contents