torch.remainder

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

Поэлементно вычисляет операцию взятия остатка по модулю в Python. Результат имеет тот же знак, что и делитель other, а его абсолютное значение меньше абсолютного значения other.

Её также можно определить через torch.div() следующим образом:

torch.remainder(a, b) == a - a.div(b, rounding_mode="floor") * b

Поддерживает трансляцию до общей формы, приведение типов, а также целочисленные значения и числа с плавающей запятой.

Примечание

Комплексные значения не поддерживаются. В некоторых случаях математически невозможно удовлетворить определению операции по модулю для комплексных чисел. О том, как обрабатывается деление на ноль, см. torch.fmod().

См. также

torch.fmod(), реализующую std::fmod из C++. Эта операция определяется через деление с округлением к нулю.

Параметры:

• input (Tensor или Scalar) – делимое
• other (Tensor или Scalar) – делитель

Именованные аргументы:

out (Tensor, необязательно) – выходной тензор.

Пример:

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