파이토치 최대값, 최소값 필터링 clamp()

파이토치 최대값, 최소값 필터링, clamp()¶

In [10]:

import torch
a = torch.FloatTensor([1, 4, 6, 8, 10, 14])
a

Out[10]:

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

• clamp(텐서, min=최소값, max=최대값)
  • 최소값보다 작은 값은 최소값으로 변경, 최대값보다 큰 값은 최대값으로 변경

In [12]:

out = torch.clamp(a, min=4, max=10)
print(out)

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