jax.lax.sqrt

Warning

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

jax.lax.sqrt¶

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

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

Parameters

x (Any) –

Return type

Any

previous

jax.lax.sort_key_val

next

jax.lax.square