Fonction torch.clamp de PyTorch
Manuel de référence torch de Pytorch
torch.clampest une fonction de PyTorch utilisée pour limiter les valeurs d'un tenseur dans un intervalle spécifié.
Définition de fonction
torch.clamp(input, min, max, out)
Exemples d’utilisation
Exemple
import torch
x = torch.tensor([-2, -1, 0, 1, 2, 3])
# Limiter entre -1 et 2
résultat = torch.clamp(x, min=-1, max=2)
print(result)
x = torch.tensor([-2, -1, 0, 1, 2, 3])
# Limiter entre -1 et 2
résultat = torch.clamp(x, min=-1, max=2)
print(result)
Le résultat de sortie est:
tensor([-1, -1, 0, 1, 2, 2])
Autres extensions