jax.lax.rsqrt

Contents

jax.lax.rsqrt#

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

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

Parameters:

x (jax.typing.ArrayLike)

Return type:

Array