當前位置: 首頁>>代碼示例 >>用法及示例精選 >>正文


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 ,並將每個商向零舍入。等效地,它會截斷商:

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

例子:

>>> 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.org大神的英文原創作品 torch.floor_divide。非經特殊聲明,原始代碼版權歸原作者所有,本譯文未經允許或授權,請勿轉載或複製。