Python: método PyTorch fmod()

El método PyTorch torch.fmod()proporciona el resto de elementos de la división por divisor. El divisor puede ser un número o un Tensor.

Sintaxis: torch.fmod(input, div, out=None)

Argumentos

• entrada: Este es el tensor de entrada.
• div: Esto puede ser un número o un tensor.
• out: El tensor de salida.

Return: Devuelve un Tensor.

Veamos este concepto con la ayuda de algunos ejemplos:
Ejemplo 1:

# Importing the PyTorch library 
import torch 
    
# A constant tensor of size n
a = torch.FloatTensor([5, 6, 7, 4])
print(a)
  
# Applying the fmod function and 
# storing the result in 'out'
out = torch.fmod(a, 3)
print(out)

Producción:

5
 6
 7
 4
[torch.FloatTensor of size 4]
 2
 0
 1
 1
[torch.FloatTensor of size 4]

Ejemplo 2:

# Importing the PyTorch library 
import torch 
    
# A constant tensor of size n
a = torch.FloatTensor([5, 6, 7, 4])
b = torch.FloatTensor([2, 3, 4, 1])
print(a, b)
  
# Applying the fmod function and 
# storing the result in 'out'
out = torch.fmod(a, b)
print(out)

Producción:

5
 6
 7
 4
[torch.FloatTensor of size 4]
 2
 3
 4
 1
[torch.FloatTensor of size 4]

 1
 0
 3
 0
[torch.FloatTensor of size 4]