torch.floor_divide

torch.floor_divide(input, other, *, out=None) → Tensor

注意

在 PyTorch 1.13 之前，torch.floor_divide() 错误地执行了截断除法。要恢复之前的 behavior，请使用 torch.div() 并设置 rounding_mode='trunc'。

计算input除以other的元素级结果，并对结果进行向下取整。

$\text{{out}}_i = \text{floor} \left( \frac{{\text{{input}}_i}}{{\text{{other}}_i}} \right)$

支持广播到统一形状、类型转换，并兼容整数和浮点数输入。

参数

• 输入 (Tensor 或 Number) – 被除数

• other (Tensor 或 Number) – 除数

关键字参数

out (Tensor, 可选) – 指定输出张量。

示例：

>>> a = torch.tensor([4.0, 3.0])
>>> b = torch.tensor([2.0, 2.0])
>>> torch.floor_divide(a, b)
tensor([2.0, 1.0])
>>> torch.floor_divide(a, 1.4)
tensor([2.0, 2.0])