Función torch.square de PyTorch

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

torch.squareEs una función de PyTorch que se utiliza para calcular el cuadrado elemento a elemento.

Definición de la función

torch.square(input, out=None)

Ejemplo de uso

Ejemplo

import torch

# Calcula el cuadrado elemento a elemento
x = torch.tensor([1.0, 2.0, 3.0])
result = torch.square(x)
print(result)

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

Otras extensiones