PyTorch torch.renorm 関数
Pytorch torch リファレンスマニュアルtorch.renormこれは PyTorch でテンソルを再正規化するための関数です。指定されたノルムに基づいてテンソルを正規化し、指定された次元の要素ノルムが指定された値を超えないようにします。
関数定義
torch.renorm(input, p, dim, maxnorm)
パラメータの説明：
input: 入力テンソルp: ノルムの次数dim: 正規化する次元maxnorm: 最大ノルム値
使用例
例
import torch
# テンソルを作成
x = torch.tensor([[2.0, 4.0, 6.0], [3.0, 6.0, 9.0]])
# dim=1 で L2 ノルム正規化を行い、最大ノルムを 1 にします
y = torch.renorm(x, p=2, dim=1, maxnorm=1)
print(y)
# テンソルを作成
x = torch.tensor([[2.0, 4.0, 6.0], [3.0, 6.0, 9.0]])
# dim=1 で L2 ノルム正規化を行い、最大ノルムを 1 にします
y = torch.renorm(x, p=2, dim=1, maxnorm=1)
print(y)
出力結果は次のとおりです：
tensor([[0.2673, 0.5345, 0.8018],
[0.2673, 0.5345, 0.8018]])
例
import torch
# テンソルを作成
x = torch.tensor([[1.0, 2.0, 3.0]])
# dim=1 で L1 ノルム正規化を行います
y = torch.renorm(x, p=1, dim=1, maxnorm=1)
print(y)
# テンソルを作成
x = torch.tensor([[1.0, 2.0, 3.0]])
# dim=1 で L1 ノルム正規化を行います
y = torch.renorm(x, p=1, dim=1, maxnorm=1)
print(y)
出力結果は次のとおりです：
tensor([[0.1667, 0.3333, 0.5000]])
その他の拡張