Skip to main content
Ctrl+K

jax.lax.sqrt#

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

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

Parameters:

x (Union[Array, ndarray, bool_, number, bool, int, float, complex]) –

Return type:

Array

previous

jax.lax.sort_key_val

next

jax.lax.square