megengine.functional.round

round(x)[源代码]

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

Rounds each element \(x_i\) of the input tensor \(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)