Функция torch.sign в PyTorch

Pytorch torch 参考手册Справочное руководство PyTorch torch

torch.signЭто функция в PyTorch, используемая для возврата знака поэлементно. Она возвращает -1 (отрицательный), 0 (ноль) или 1 (положительный).

Определение функции

torch.sign(input, out=None)

Примеры использования

Пример

import torch

# Вернуть знак поэлементно
x = torch.tensor([-2.0, 0.0, 3.0])
result = torch.sign(x)
print(result)

Pytorch torch 参考手册Справочное руководство PyTorch torch

Другие расширения