jax.lax.sqrt¶

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

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

Parameters

x (Any) –

Return type

Any