Función torch.div de PyTorch
Manual de referencia de Pytorch torch
torch.divEs una función en PyTorch que se utiliza para realizar la división elemento a elemento.
Definición de la función
torch.div(input, other, rounding_mode, out)
Ejemplo de uso
Ejemplo
import torch
a = torch.tensor([10, 20, 30])
b = torch.tensor([2, 4, 5])
# División elemento a elemento
c = torch.div(a, b)
print(c)
a = torch.tensor([10, 20, 30])
b = torch.tensor([2, 4, 5])
# División elemento a elemento
c = torch.div(a, b)
print(c)
El resultado de salida es:
tensor([5., 5., 6.])
Otras extensiones