torch_gcd {torch} | R Documentation |
Gcd
torch_gcd(self, other)
self |
(Tensor) the input tensor. |
other |
(Tensor) the second input tensor |
Computes the element-wise greatest common divisor (GCD) of input
and other
.
Both input
and other
must have integer types.
This defines gcd(0, 0) = 0
.
if (torch_is_installed()) {
if (torch::cuda_is_available()) {
a <- torch_tensor(c(5, 10, 15), dtype = torch_long(), device = "cuda")
b <- torch_tensor(c(3, 4, 5), dtype = torch_long(), device = "cuda")
torch_gcd(a, b)
c <- torch_tensor(c(3L), device = "cuda")
torch_gcd(a, c)
}
}