[PyTorch] torch.clamp 함수

* Pytorch 공식 페이지 설명을 참고했습니다.

torch.clamp — PyTorch 1.10.0 documentation

 

Clamp 함수는 입력으로 들어오는 모든 값들을 [min, max]의 범위 안으로 조정해주는 역할을 합니다.

 

조정해주는 방법은 다음 식과 같습니다. 

input 값과 min 값 중 큰 값 = a

a와 max 값 중 작은 값 = 최종 반환 값

 

즉, 입력 값이 min < x(input) < max이면 x가 그대로 나오지만

min 보다 작으면 min 값이, max 보다 크면 max 값이 나오는 구조입니다!

 

참고로 min 값과 max를 꼭 둘다 설정할 필요는 없습니다.

min을 설정하지 않으면(min = None), lower bound가 존재하지 않고, 

max를 설정하지 않으면(max = None), upper bound가 존재하지 않으니 필요에 따라 설정하시면 됩니다.

 

 

* 공식 문서 사용법 예제

a = torch.randn(4) 
# a = tensor([-1.7120,  0.1734, -0.0478, -0.0922])

torch.clamp(a, min=-0.5, max=0.5) 
# a = tensor([-0.5000,  0.1734, -0.0478, -0.0922])