dpnp.sqrt

dpnp.sqrt(x, /, out=None, *, order='K', where=True, dtype=None, subok=True, **kwargs)

Return the non-negative square-root of an array, element-wise.

For full documentation refer to numpy.sqrt.

Returns:

out – An array of the same shape as x, containing the positive square-root of each element in x. If any element in x is complex, a complex array is returned (and the square-roots of negative reals are calculated). If all of the elements in x are real, so is y, with negative elements returning nan.

Return type:

dpnp.ndarray

Limitations

Input array is supported as either dpnp.ndarray or dpctl.tensor.usm_ndarray. Parameter out is supported as class:dpnp.ndarray, class:dpctl.tensor.usm_ndarray or with default value None. Parameters where, dtype and subok are supported with their default values. Otherwise the function will be executed sequentially on CPU. Input array data types are limited by supported DPNP Data types.

See also

dpnp.cbrt

Return the cube-root of an array, element-wise.

dpnp.rsqrt

Return the reciprocal square-root of an array, element-wise.

Examples

>>> import dpnp as np
>>> x = np.array([1, 4, 9])
>>> np.sqrt(x)
array([1., 2., 3.])

>>> x2 = np.array([4, -1, np.inf])
>>> np.sqrt(x2)
array([ 2., nan, inf])