torch.fmod¶
- torch.fmod(input, other, *, out=None) Tensor ¶
逐元素应用 C++ 的 std::fmod。结果与被除数
input
的符号相同,其绝对值小于other
的绝对值。此函数可以根据
torch.div()
定义为torch.fmod(a, b) == a - a.div(b, rounding_mode="trunc") * b
注意
当除数为零时,对于 CPU 和 GPU 上的浮点数类型返回
NaN
;对于 CPU 上的整数除以零引发RuntimeError
;GPU 上的整数除以零可能会返回任何值。注意
不支持复数输入。在某些情况下,用复数来满足模运算的定义在数学上是不可能的。
另请参见
torch.remainder()
,它实现了 Python 的模运算符。这个定义是使用向下取整的结果来进行除法的。示例
>>> 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])