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(), который реализует C++’s std::fmod. Этот вариант определяется через деление с округлением к нулю.

Параметры

• input (Tensor или Скаляр) – делимое
• other (Tensor или Скаляр) – делитель

Ключевые аргументы

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