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 ，包含 \(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)
Element-wise square root:
>>> x = Tensor([1, 4, 9, 16]) >>> F.sqrt(x) Tensor([1. 2. 3. 4.], device=xpux:0)