Función torch.sign de PyTorch

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

torch.signEs una función en PyTorch que se utiliza para devolver el signo elemento por elemento. Devuelve -1 (negativo), 0 (cero) o 1 (positivo).

Definición de la función

torch.sign(input, out=None)

Ejemplo de uso

Ejemplo

import torch

# Devuelve el signo elemento por elemento
x = torch.tensor([-2.0, 0.0, 3.0])
result = torch.sign(x)
print(result)

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

Otras extensiones