square

square(x, out=None, **kwargs)

Return the element-wise square of the input.

Parameters:

x : ndarray or scalar

The values whose squares are required.

out : ndarray or None, optional

A location into which the result is stored. If provided, it must have the same shape as the input. If not provided or None, a freshly-allocated array is returned.

Returns:

y : ndarray or scalar

Output array is same shape and type as x. This is a scalar if x is a scalar.

Notes

The output ndarray has the same ctx as the input ndarray. This function differs from the original numpy.square in the following aspects: - Only support ndarray and scalar now. - where argument is not supported. - Complex input is not supported.

Examples

>>> np.square(2.)
4.0
>>> x = np.array([1, 2., -1])
>>> np.square(x)
array([1., 4., 1.])