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)