Функция torch.heaviside в PyTorch
Справочное руководство Pytorch torch
torch.heavisideЭто функция в PyTorch, используемая для вычисления ступенчатой функции Хевисайда. Она возвращает 1, если входное значение больше 0, 0.5, если равно 0, и 0, если меньше 0.
Определение функции
torch.heaviside(input, values)
Пример использования
Пример
import torch
# Создать входной тензор
input = torch.tensor([-1.0, -0.5, 0.0, 0.5, 1.0])
# Укажите возвращаемое значение, когда входное значение равно 0
values = torch.tensor([0.5])
# Вычислите ступенчатую функцию Хевисайда
x = torch.heaviside(input, values)
print(x)
# Создать входной тензор
input = torch.tensor([-1.0, -0.5, 0.0, 0.5, 1.0])
# Укажите возвращаемое значение, когда входное значение равно 0
values = torch.tensor([0.5])
# Вычислите ступенчатую функцию Хевисайда
x = torch.heaviside(input, values)
print(x)
Другие расширения