megengine.functional.floor

floor(x)[源代码]

逐元素 \(\lfloor x \rfloor\) 操作。

对输入张量 \(x\) 的每一个元素 \(x_i\) 取值为不超过 \(x_i\) 的最大整数值。

参数

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。

实际案例

>>> F.floor(1.5)
Tensor(1.0, device=xpux:0)

逐元素取floor操作:

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