Función PyTorch torch.divide
Manual de referencia de PyTorch torch
torch.divideEs la función de PyTorch para la división elemento a elemento (equivalente a div).
Definición de la función
torch.divide(input, other, *, rounding_mode=None, out=None)
Ejemplo de uso
Ejemplo
import torch
# Ejemplo de división elemento a elemento
x = torch.tensor([10.0, 20.0, 30.0])
y = torch.tensor([2.0, 4.0, 5.0])
result = torch.divide(x, y)
print(result) # tensor([5.0000, 5.0000, 6.0000])
# Ejemplo de división elemento a elemento
x = torch.tensor([10.0, 20.0, 30.0])
y = torch.tensor([2.0, 4.0, 5.0])
result = torch.divide(x, y)
print(result) # tensor([5.0000, 5.0000, 6.0000])
Otras extensiones