PyTorch torch.hypot 関数
torch.hypotPyTorch において直角三角形の斜辺の長さを計算する関数であり、すなわち sqrt(a^2 + b^2) を計算します。
関数定義
torch.hypot(input, other, out=None)
使用例
実例
import torch
# 2つの直角辺を作成
a = torch.tensor([3.0, 5.0])
b = torch.tensor([4.0, 12.0])
# 斜辺の長さを計算
c = torch.hypot(a, b)
print("斜辺:", c)
# 2つの直角辺を作成
a = torch.tensor([3.0, 5.0])
b = torch.tensor([4.0, 12.0])
# 斜辺の長さを計算
c = torch.hypot(a, b)
print("斜辺:", c)
出力結果は次のとおりです:
斜边: tensor([ 5., 13.])
その他の拡張
Pytorch torch リファレンスマニュアル