dpnp.square

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

Computes x_i**2 (or x_i*x_i) for each element x_i of input array x.

For full documentation refer to numpy.square.

Parameters:

  • x ({dpnp.ndarray, usm_ndarray}) – Input array.

  • out ({None, dpnp.ndarray}, optional) – Output array to populate. Array must have the correct shape and the expected data type.

  • order ({"C", "F", "A", "K"}, optional) – Memory layout of the newly output array, if parameter out is None. Default: “K”.

Returns:

out – An array containing the element-wise square results.

Return type:

dpnp.ndarray

Limitations

Parameters where and subok are supported with their default values. Otherwise NotImplementedError exception will be raised.

See also

dpnp..linalg.matrix_power

Raise a square matrix to the (integer) power n.

dpnp.sqrt

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

dpnp.power

First array elements raised to powers from second array, element-wise.

Examples

>>> import dpnp as np
>>> x = np.array([-1j, 1])
>>> np.square(x)
array([-1.+0.j,  1.+0.j])