jax.lax.rsqrt

Warning

This page was created from a pull request (#9655).

jax.lax.rsqrt¶

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

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

Parameters

x (Any) –

Return type

Any

previous

jax.lax.round

next

jax.lax.scatter