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