Python PyTorch fmod用法及代碼示例

本文簡要介紹python語言中 torch.fmod 的用法。

用法:

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

參數：

• input(Tensor) -股息

• other(Tensor或者標量) -除數

關鍵字參數：

out(Tensor,可選的) -輸出張量。

對浮點張量應用 C++ 的std::fmod，對整數張量應用模運算。結果與被除數 input 具有相同的符號，並且其絕對值小於 other 的絕對值。

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

注意

當除數為零時，為 CPU 和 GPU 上的浮點 dtypes 返回 NaN；在 CPU 上引發 RuntimeError 整數除以零； GPU 上的整數除以零可能會返回任何值。

注意

不支持複雜輸入。在某些情況下，用複數滿足模運算的定義在數學上是不可能的。

例子：

>>> torch.fmod(torch.tensor([-3., -2, -1, 1, 2, 3]), 2)
tensor([-1., -0., -1.,  1.,  0.,  1.])
>>> torch.fmod(torch.tensor([1, 2, 3, 4, 5]), -1.5)
tensor([1.0000, 0.5000, 0.0000, 1.0000, 0.5000])