megengine.functional.round

round(x)

逐元素的 \(\operatorname{round}(x)\) 函数.

Rounds each element \(x_i\) of the input tebsor \(x\) to the nearest integer-valued number.

参数:

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

返回:

一个包含 \(x\) 中每个元素的取整结果的张量。返回张量一定具有和 \(x\) 相同的数据类型。

特殊的案例

如果 \(x_i\) 已经是整数，那么结果是 \(x_i\) 。

对于浮点数的操作，

• 若 \(x_i\) 是 +infinity，则结果为 +infinity。

• 如果 \(x_i\) 是 -infinity ，那么结果是 -infinity 。

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

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

• 如果 \(x_i\) 是 NaN，那么结果是 NaN。

• 如果两个整数同样接近 \(x_i\)，结果是最接近 \(x_i\) 的偶数整数。

实际案例

>>> F.round(1.5)
Tensor(2.0, device=xpux:0)

逐元素 rounding：

>>> x = Tensor([1.5, 2.5, 3.5, 4.5])
>>> F.round(x)
Tensor([2. 3. 4. 5.], device=xpux:0)