Fonction torch.clip de PyTorch
Manuel de référence PyTorch torch
torch.clipest une fonction de PyTorch utilisée pour limiter les valeurs d’un tenseur à une plage spécifiée.
Définition de fonction
torch.clip(input, min, max, out=None)
Exemples d'utilisation
Exemple
import torch
# Créer un tenseur
x = torch.tensor([1.0, 2.0, 3.0, 4.0, 5.0])
# Limiter les valeurs à la plage [2, 4]
y = torch.clip(x, 2, 4)
print(y)
# Créer un tenseur
x = torch.tensor([1.0, 2.0, 3.0, 4.0, 5.0])
# Limiter les valeurs à la plage [2, 4]
y = torch.clip(x, 2, 4)
print(y)
Le résultat de sortie est:
tensor([2., 2., 3., 4., 4.])
Autres extensions