파이토치 clamp 최소값, 최대값 고정

>

파이썬 파이토치 최소값, 최대값 고정¶

• clamp는 텐서의 원소들의 값들을 지정한 범위 내로 제한하는 매서드 입니다

In [1]:

# 예시 텐서 생성
import torch
a = torch.FloatTensor([1, 4, 6, 8, 10, 14])
a

Out[1]:

tensor([ 1.,  4.,  6.,  8., 10., 14.])

In [2]:

# 4이하는 4 10 이상은 10으로 고정
out = torch.clamp(a, min=4, max=10)
out

Out[2]:

tensor([ 4.,  4.,  6.,  8., 10., 10.])

• 최소값 혹은 최대값 중 하나만 고정하는 기능도 가능합니다.

In [3]:

# 최소값만 고정
min_only = torch.clamp(a, min=4)
min_only

Out[3]:

tensor([ 4.,  4.,  6.,  8., 10., 14.])

In [4]:

# 최소값만 고정
man_only = torch.clamp(a, max=10)
man_only

Out[4]:

tensor([ 1.,  4.,  6.,  8., 10., 10.])