megengine.functional.abs

abs(x)[源代码]

将函数 \(\operatorname{abs}(x)\) 应用于每一个元素。

计算输入张量 \(x\) 中每个元素 \(x_i\) 的绝对值。(即,结果与 x 中的对应元素具有相同的幅度,但是符号为正)。

参数

x – 输入张量。 应该有一个数字数据类型。

返回

tensor,该 tensor 中的元素为对 \(x\) 中每个元素求绝对值所得到的结果。返回的 tensor 必须与 \(x\) 具有相同的数据类型。

特殊的案例

对于浮点数的操作,

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

  • \(x_i\)-0,则结果为 +0

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

实际案例

>>> F.abs(-2)
Tensor(2, dtype=int32, device=xpux:0)

逐元素计算绝对值:

>>> x = Tensor([1, -2, 3, -4, 5])
>>> F.abs(x)
Tensor([1 2 3 4 5], dtype=int32, device=xpux:0)