mindspore.numpy.sqrt

mindspore.numpy.sqrt(x, dtype=None)[源代码]

逐元素返回数组的非负平方根。

说明

不支持NumPy参数 out 、 where 、 casting 、 order 、 subok 、 signature 、 extobj 。在 GPU 上，支持的 dtype 为 np.float16 和 np.float32。

参数：

  • x (Tensor) - 需计算平方根的值。

  • dtype (mindspore.dtype, 可选) - 默认值： None 。覆盖输出Tensor的dtype。

返回：

Tensor或标量，与 x shape相同的数组，包含 x 中每个元素的正平方根。 对于负元素，返回 nan。如果 x 是标量，则返回标量。

支持平台：

Ascend GPU CPU

样例：

>>> import mindspore.numpy as np
>>> x = np.arange(6).reshape(2, 3).astype('float32')
>>> x_squared = np.square(x)
>>> output = np.sqrt(x_squared)
>>> print(output)
[[ 0. 1. 2.]
[ 3. 4. 5.]]