jax.lax.rsqrt#

jax.lax.rsqrt(x)[source]#

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

Parameters

x (Union[Array, ndarray, bool_, number, bool, int, float, complex]) –

Return type

Array