Función torch.clip de PyTorch
Manual de referencia de PyTorch torch
torch.clipEs una función en PyTorch utilizada para limitar los valores de un tensor a un rango especificado.
Definición de la función
torch.clip(input, min, max, out=None)
Ejemplo de uso
Ejemplo
import torch
# Crear tensor
x = torch.tensor([1.0, 2.0, 3.0, 4.0, 5.0])
# Limitar los valores al rango [2, 4]
y = torch.clip(x, 2, 4)
print(y)
# Crear tensor
x = torch.tensor([1.0, 2.0, 3.0, 4.0, 5.0])
# Limitar los valores al rango [2, 4]
y = torch.clip(x, 2, 4)
print(y)
El resultado de salida es:
tensor([2., 2., 3., 4., 4.])
Otras extensiones