mindspore.numpy.sqrt

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

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

说明

不支持NumPy参数 outwherecastingordersuboksignatureextobj 。在 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.]]