Función torch.sgn de PyTorch

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

torch.sgnEs una función en PyTorch que devuelve el vector de signos elemento por elemento. Para números complejos, devuelve el número complejo unitario.

Definición de la función

torch.sgn(input, out=None)

Ejemplo de uso

Ejemplo

import torch

# Devuelve el vector de signos elemento por elemento
x = torch.tensor([-2.0, 0.0, 3.0])
result = torch.sgn(x)
print(result)

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

Otras extensiones