Python PyTorch floor_divide用法及代码示例

本文简要介绍python语言中 torch.floor_divide 的用法。

用法:

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

参数：

• input(Tensor或者数字) -股息

• other(Tensor或者数字) -除数

关键字参数：

out(Tensor,可选的) -输出张量。

警告

torch.floor_divide() 已弃用，并将在未来的 PyTorch 版本中删除。它的名字用词不当，因为它实际上将商四舍五入为零，而不是取下限。要保持当前行为，请使用 torch.div() 和 rounding_mode='trunc' 。要实际执行楼层划分，请使用 torch.div() 和 rounding_mode='floor' 。

按元素计算 input 除以 other ，并将每个商向零舍入。等效地，它会截断商：

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

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

例子：

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