快捷方式

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,并对结果进行向下取整。

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

支持广播到公共形状、类型提升以及整数和浮点数输入。

参数
  • input (张量数字) – 被除数

  • other (张量数字) – 除数

关键字参数

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

示例

>>> 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])

文档

访问 PyTorch 的全面开发者文档

查看文档

教程

获取针对初学者和高级开发人员的深入教程

查看教程

资源

查找开发资源并获得问题的解答

查看资源