PyTorch torch.heaviside-Funktion
PyTorch torch Referenzhandbuch
torch.heavisideEs ist eine Funktion in PyTorch, die zur Berechnung der Heaviside-Schrittfunktion verwendet wird. Sie gibt 1 zurück, wenn der Eingabewert größer als 0 ist, 0.5, wenn er gleich 0 ist, und 0, wenn er kleiner als 0 ist.
Funktionsdefinition
torch.heaviside(input, values)
Anwendungsbeispiel
Beispiel
import torch
# Eingabe-Tensor erstellen
input = torch.tensor([-1.0, -0.5, 0.0, 0.5, 1.0])
# Geben Sie den Rückgabewert an, wenn die Eingabe 0 ist
values = torch.tensor([0.5])
# Berechnen Sie die Heaviside-Schrittfunktion
x = torch.heaviside(input, values)
print(x)
# Eingabe-Tensor erstellen
input = torch.tensor([-1.0, -0.5, 0.0, 0.5, 1.0])
# Geben Sie den Rückgabewert an, wenn die Eingabe 0 ist
values = torch.tensor([0.5])
# Berechnen Sie die Heaviside-Schrittfunktion
x = torch.heaviside(input, values)
print(x)
Andere Erweiterungen