torch.gcd

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

Вычисляет поэлементное наибольшее общее делитель (НОД) input и other.

И input и other должны иметь целочисленные типы.

Примечание

Это определяет $gcd(0, 0) = 0$.

Параметры

• input (Tensor) – входной тензор.
• other (Tensor) – второй входной тензор

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

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

Пример:

>>> a = torch.tensor([5, 10, 15])
>>> b = torch.tensor([3, 4, 5])
>>> torch.gcd(a, b)
tensor([1, 2, 5])
>>> c = torch.tensor([3])
>>> torch.gcd(a, c)
tensor([1, 1, 3])