Функция torch.sgn в PyTorch
Справочное руководство Pytorch torch
torch.sgn— это функция в PyTorch, используемая для поэлементного возврата вектора знаков. Для комплексных чисел она возвращает единичное комплексное число.
Определение функции
torch.sgn(input, out=None)
Пример использования
Пример
import torch
# Поэлементно вернуть вектор знаков
x = torch.tensor([-2.0, 0.0, 3.0])
result = torch.sgn(x)
print(result)
# Поэлементно вернуть вектор знаков
x = torch.tensor([-2.0, 0.0, 3.0])
result = torch.sgn(x)
print(result)
Другие расширения