numpy.square

numpy.square(x[, out]) = <ufunc 'square'>

返回输入的元素平方。

参数：

x：array_like

输入数据。

返回：

out：ndarray

元素方式x * x，具有相同的形状，dtype为x。如果x是标量，则返回标量。

也可以看看

numpy.linalg.matrix_power，sqrt，power

例子

>>> np.square([-1j, 1])
array([-1.-0.j,  1.+0.j])