PyTorch torch.view_as_complex 関数
torch.view_as_complexPyTorch において、実数テンソルのビューを複素数テンソルに変換するための関数です。最後の次元を2つの実数値(実部と虚部)から単一の複素数値に変換します。
関数定義
torch.view_as_complex(input)
使用例
例
import torch
# 実数テンソルを作成。最後の次元には2つの要素が必要です
x = torch.randn(3, 2)
# 複素数ビューに変換
y = torch.view_as_complex(x)
print("元の実数テンソルの形状:", x.shape)
print("複素数テンソルの形状:", y.shape)
print("複素数テンソル:", y)
# 実数テンソルを作成。最後の次元には2つの要素が必要です
x = torch.randn(3, 2)
# 複素数ビューに変換
y = torch.view_as_complex(x)
print("元の実数テンソルの形状:", x.shape)
print("複素数テンソルの形状:", y.shape)
print("複素数テンソル:", y)
出力結果は次のとおりです:
原始实数张量形状: torch.Size([3, 2]) 复数张量形状: torch.Size([3]) 复数张量: tensor([ 1.2345+0.5678j, -0.3456+1.2345j, 2.3456-0.9876j])
その他の拡張
Pytorch torch リファレンスマニュアル