Función torch.sqrt de PyTorch
Manual de referencia de Pytorch torch
torch.sqrtEs una función en PyTorch utilizada para calcular la raíz cuadrada elemento a elemento de un tensor.
Definición de la función
torch.sqrt(input, out=None)
Ejemplo de uso
Ejemplo
import torch
x = torch.tensor([4.0, 9.0, 16.0, 25.0])
y = torch.sqrt(x)
print(y)
x = torch.tensor([4.0, 9.0, 16.0, 25.0])
y = torch.sqrt(x)
print(y)
El resultado de salida es:
tensor([2., 3., 4., 5.])
Otras extensiones