torch.clamp

torch.clamp(input, min, max, out=None) → Tensor

Clamp all elements in input into the range [ min, max ] and return a resulting tensor:

If input is of type FloatTensor or DoubleTensor, args min and max must be real numbers, otherwise they should