PyTorch torch.atan2 関数
Pytorch torch リファレンスマニュアルtorch.atan2PyTorch において、要素ごとに2次元の逆正接（arctangent2）を計算するための関数です。2つの座標値 (y, x) に基づいて角度を計算し、戻り値の単位はラジアンで、範囲は [-pi, pi] です。
関数定義
torch.atan2(input, other, out=None)
パラメータの説明
input: 分子テンソル（y座標）other: 分母テンソル（x座標）out: 出力テンソル（オプション）
使用例
実例
import torch
# テンソルを作成
y = torch.tensor([0.0, 1.0, -1.0, 1.0])
x = torch.tensor([1.0, 0.0, 0.0, -1.0])
# 2次元の逆正接を計算
result = torch.atan2(y, x)
print(result)
# テンソルを作成
y = torch.tensor([0.0, 1.0, -1.0, 1.0])
x = torch.tensor([1.0, 0.0, 0.0, -1.0])
# 2次元の逆正接を計算
result = torch.atan2(y, x)
print(result)
出力結果は次のとおりです：
tensor([ 0.0000, 1.5708, -1.5708, 2.3562])
その他の拡張