Función PyTorch torch.cos
Manual de referencia de Pytorch torch
torch.cosEs la función de PyTorch para calcular el coseno elemento a elemento.
Definición de la función
torch.cos(input, out=None)
Ejemplo de uso
Ejemplo
import torch
x = torch.tensor([0, torch.pi/2, torch.pi, 3*torch.pi/2])
print(torch.cos(x))
x = torch.tensor([0, torch.pi/2, torch.pi, 3*torch.pi/2])
print(torch.cos(x))
El resultado de salida es:
tensor([ 1.0000e+00, -8.7423e-08, -1.0000e+00, -8.7423e-08])
Otras extensiones