PyTorch配列Tesonrを任意の最小値・最大値に収めるtorch.clamp

torch.clampの使い方 第一引数inputに処理する配列Tesonr、第二引数minに最小値、第三引数maxに最大値を指定する。 import torch a = torch.tensor([-1.0, -0.4, 0.0, 0.3, 0.9]) print(a) # tensor([-1.0000, -0.4000, 0.0000, 0.3000, 0.9000]) print(torch.clamp(input=a, min=-0.5, max=0.5)) # tensor([-0.5000, -0.4000, 0.0000, 0.3000, 0.5000])

[stealthinu]

torchのtensorをある値まででサチらせたいと思ったらclampという便利な関数があった。こりゃ楽～。

[take-tech-engineer]

PyTorch配列Tesonrを任意の最小値・最大値に収めるclamp関数について、まとめました。Numpyの場合、clip関数で同等のことができそうです。