Función torch.pow de PyTorch

Pytorch torch 参考手册Manual de referencia de Pytorch torch

torch.powEs una función en PyTorch utilizada para calcular la potenciación elemento a elemento.

Definición de la función

torch.pow(input, exponent, out)

Ejemplo de uso

Ejemplo

import torch

# Calcular la potencia
x = torch.tensor([2, 3, 4])
result = torch.pow(x, 2)
print(result)

# También se puede usar el exponente como primer parámetro
result2 = torch.pow(2, x)
print(result2)

Pytorch torch 参考手册Manual de referencia de Pytorch torch

Otras extensiones