PyTorch torch.heaviside función
Pytorch torch manual de referencia
torch.heavisideEs una función en PyTorch utilizada para calcular la función escalón de Heaviside. Cuando el valor de entrada es mayor que 0, devuelve 1; igual a 0, devuelve 0.5; menor que 0, devuelve 0.
Definición de la función
torch.heaviside(input, values)
Ejemplo de uso
Ejemplo
import torch
# Crear tensor de entrada
input = torch.tensor([-1.0, -0.5, 0.0, 0.5, 1.0])
# Especificar el valor de retorno cuando la entrada es 0
values = torch.tensor([0.5])
# Calcular la función escalón de Heaviside
x = torch.heaviside(input, values)
print(x)
# Crear tensor de entrada
input = torch.tensor([-1.0, -0.5, 0.0, 0.5, 1.0])
# Especificar el valor de retorno cuando la entrada es 0
values = torch.tensor([0.5])
# Calcular la función escalón de Heaviside
x = torch.heaviside(input, values)
print(x)
Otras extensiones