Función torch.exp2 de PyTorch
Manual de referencia de PyTorch torch
torch.exp2Es la función de PyTorch para calcular la potencia de 2 elemento a elemento.
Definición de la función
torch.exp2(input, *, out=None)
Ejemplo de uso
Ejemplo
import torch
# Ejemplo de potencia de 2
x = torch.tensor([0, 1, 2, 3, 4], dtype=torch.float32)
result = torch.exp2(x)
print(result) # tensor([1., 2., 4., 8., 16.])
# Ejemplo de potencia de 2
x = torch.tensor([0, 1, 2, 3, 4], dtype=torch.float32)
result = torch.exp2(x)
print(result) # tensor([1., 2., 4., 8., 16.])
Otras extensiones