dpnp.square

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

Squares 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, usm_ndarray}, optional) -- Output array to populate. Array must have the correct shape and the expected data type. Default: None.

  • 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 squares of x. The data type of the returned array is determined by the Type Promotion Rules.

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])