Pythonのtorch.clamp()関数の使い方は何ですか？

1年 ago
結衣, 春花
1 minute

torch.clamp()関数は、入力テンソル内の要素を指定した範囲内に制限するために使用されます。

関数の構文は以下の通りです：

torch.clamp(input, min, max, out=None)

引数の説明:

  1. 入力：テンソル。
  2. 最小値：この値未満の要素は、この値に置き換えられます。
  3. max：その値を超える要素は、その値に置き換えられます。
  4. 出力：出力結果を保存するテンソル。

利用例：

import torch

x = torch.randn(5)  # 创建一个包含5个随机数的张量
print(x)  # 打印原始张量

# 使用clamp函数将张量的元素限制在-0.5到0.5之间
y = torch.clamp(x, -0.5, 0.5)
print(y)  # 打印限制后的张量

出力結果の例：

tensor([ 0.0849, -0.2706,  0.7244,  0.0921,  0.6237])
tensor([ 0.0849, -0.2706,  0.5000,  0.0921,  0.5000])

上記の例では、まず5つのランダムな数値を持つテンソルxが作成され、その後clamp関数を使用してそのテンソルの要素を-0.5から0.5の間に制限し、その結果をテンソルyに格納します。最後に元のテンソルと制限されたテンソルを表示します。テンソルyの要素はすべて-0.5から0.5の間に制限されていることが確認できます。

bannerAds