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