megengine.functional.sqrt#

sqrt(x)[源代码]#

元素 \(\operatorname{sqrt}(x)\) 函数

计算输入 Tensor \(x\) 中每个元素 \(x_i\) 的平方根。取整后,每个计算结果必须与 infinitely precise result(IEEE 754 标准) 保持一致。

函数的定义域为 [0, +infinity] , 值域为 [0, +infinity]

参数:

x (Tensor) – 输入 tensor。应该具有 float 数据类型。

返回类型:

Tensor

返回:

一个 Tensor ,包含 \(x\) 中每个元素的平方根计算结果。返回的 Tensor 将会使用浮点数据类型,具体见 类型提升规则

特殊的案例

对于浮点数的操作,

  • \(x_i\)NaN,则结果为 NaN

  • 如果 \(x_i\) 小于 0, 结果记为 NaN

  • 如果 \(x_i\)+0 ,那么结果是 +0

  • 如果 \(x_i\)-0 ,那么结果是 -0

  • \(x_i\)+infinity,则结果为 +infinity

实际案例

>>> F.sqrt(4)
Tensor(2.0, device=xpux:0)

逐元素取平方根:

>>> x = Tensor([1, 4, 9, 16])
>>> F.sqrt(x)
Tensor([1. 2. 3. 4.], device=xpux:0)