Fonction torch.heaviside de PyTorch
Manuel de référence Pytorch torch
torch.heavisideIl s'agit d'une fonction de PyTorch utilisée pour calculer la fonction échelon de Heaviside. Elle renvoie 1 lorsque la valeur d'entrée est supérieure à 0, 0.5 lorsqu'elle est égale à 0, et 0 lorsqu'elle est inférieure à 0.
Définition de fonction
torch.heaviside(input, values)
Exemple d'utilisation
Exemple
import torch
# Créer un tenseur d'entrée
input = torch.tensor([-1.0, -0.5, 0.0, 0.5, 1.0])
# Spécifier la valeur de retour lorsque l'entrée est 0
values = torch.tensor([0.5])
# Calculer la fonction échelon de Heaviside
x = torch.heaviside(input, values)
print(x)
# Créer un tenseur d'entrée
input = torch.tensor([-1.0, -0.5, 0.0, 0.5, 1.0])
# Spécifier la valeur de retour lorsque l'entrée est 0
values = torch.tensor([0.5])
# Calculer la fonction échelon de Heaviside
x = torch.heaviside(input, values)
print(x)
Autres extensions