jax.lax.rsqrt

jax.lax.rsqrt(x)[source]

Elementwise reciprocal square root: :math:`1 over sqrt{x}.

Parameters

x (Any) –

Return type

Any