PyTorch torch.clip 関数
Pytorch torch リファレンスマニュアルtorch.clipPyTorch において、テンソル内の値を指定した範囲にクリップするための関数です。
関数定義
torch.clip(input, min, max, out=None)
使用例
実例
import torch
# テンソルを作成
x = torch.tensor([1.0, 2.0, 3.0, 4.0, 5.0])
# 値を [2, 4] の範囲にクリップする
y = torch.clip(x, 2, 4)
print(y)
# テンソルを作成
x = torch.tensor([1.0, 2.0, 3.0, 4.0, 5.0])
# 値を [2, 4] の範囲にクリップする
y = torch.clip(x, 2, 4)
print(y)
出力結果は次のとおりです：
tensor([2., 2., 3., 4., 4.])
その他の拡張