Función torch.clamp de PyTorch
Manual de referencia de torch de Pytorch
torch.clampEs una función en PyTorch que se utiliza para limitar los valores de un tensor a un rango especificado.
Definición de la función
torch.clamp(input, min, max, out)
Ejemplo de uso
Ejemplo
import torch
x = torch.tensor([-2, -1, 0, 1, 2, 3])
# Limitar entre -1 y 2
result = torch.clamp(x, min=-1, max=2)
print(result)
x = torch.tensor([-2, -1, 0, 1, 2, 3])
# Limitar entre -1 y 2
result = torch.clamp(x, min=-1, max=2)
print(result)
El resultado de salida es:
tensor([-1, -1, 0, 1, 2, 2])
Otras extensiones