mindspore.numpy.square
- mindspore.numpy.square(x, dtype=None)[source]
Returns the element-wise square of the input.
Note
Numpy arguments out, where, casting, order, subok, signature, and extobj are not supported. On GPU, the supported dtypes are np.float16 and np.float32.
- Parameters
x (Tensor) – Input data.
dtype (
mindspore.dtype
, optional) – defaults to None. Overrides the dtype of the output Tensor.
- Returns
Tensor or scalar, element-wise
x*x
, of the same shape and dtype as x. This is a scalar if x is a scalar..
- Supported Platforms:
Ascend
GPU
CPU
Examples
>>> import mindspore.numpy as np >>> x = np.square(np.arange(6).reshape(2, 3).astype('float32')) >>> print(x) [[ 0. 1. 4.] [ 9. 16. 25.]]