torch.floor_divide

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

注意

在 PyTorch 1.13 之前，torch.floor_divide() 錯誤地執行了截斷除法。若要還原先前的行為，請使用 torch.div() 並搭配 rounding_mode='trunc'。

計算 input 除以 other 的結果，逐元素計算，並將結果向下取整。

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

支援廣播到通用形狀、類型提升，以及整數和浮點數輸入。

參數

• input (Tensor 或 Number) – 被除數

• other (Tensor 或 Number) – 除數

關鍵字參數

out (Tensor, optional) – 輸出張量。

範例

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