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