[PyTorch] torch.clamp()

qw4735·2023년 1월 6일

torch.clamp(input, min=None, max=None) -> Tensor

  • clamp(clamps) : 고정시키다.
  • Tensor.clamp_(min=None, max=None)은 torch.clamp()로 대체됨.
model.sigma.data.clamp_(0.01, 1.0)  # model.sigma.data의 범위를 0.01에서 1사이로 바꿈
# torch.clamp(model.sigma.data, min=0.01, max=1.0)로 바뀌어야 함.

< 공식문서 예시 >

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

텐서 a의 범위를 -0.5에서 0.5로 고정시켜라.

파이토치 공식문서 torch.clamp

0개의 댓글