jax.lax.sqrt

jax.lax.sqrt(x)[source]

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

Parameters

x (Any) –

Return type

Any