PyTorch torch.angle 関数
Pytorch torch リファレンスマニュアルtorch.anglePyTorch において、複素数テンソルの位相角（argument）を返す関数です。戻り値の単位はラジアンで、範囲は [-pi, pi] の間です。
関数定義
torch.angle(input, out=None)
使用例
例
import torch
# 複素数テンソルを作成
x = torch.tensor([1+1j, 1+0j, 0+1j, -1+0j])
# 位相角を返す
result = torch.angle(x)
print(result)
# 複素数テンソルを作成
x = torch.tensor([1+1j, 1+0j, 0+1j, -1+0j])
# 位相角を返す
result = torch.angle(x)
print(result)
出力結果は次のとおりです：
tensor([ 0.7854, 0.0000, 1.5708, 3.1416])
その他の拡張