PyTorch torch.clip-Funktion
PyTorch torch-Referenzhandbuch
torch.clipist eine Funktion in PyTorch, die verwendet wird, um die Werte in einem Tensor auf einen angegebenen Bereich zu begrenzen.
Funktionsdefinition
torch.clip(input, min, max, out=None)
Anwendungsbeispiele
Beispiel
import torch
# Tensor erstellen
x = torch.tensor([1.0, 2.0, 3.0, 4.0, 5.0])
# Werte auf den Bereich [2, 4] begrenzen
y = torch.clip(x, 2, 4)
print(y)
# Tensor erstellen
x = torch.tensor([1.0, 2.0, 3.0, 4.0, 5.0])
# Werte auf den Bereich [2, 4] begrenzen
y = torch.clip(x, 2, 4)
print(y)
Das Ausgabeergebnis ist:
tensor([2., 2., 3., 4., 4.])
Andere Erweiterungen