PyTorch torch.view_as_complex 関数


Pytorch torch 参考手册Pytorch torch リファレンスマニュアル

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)

出力結果は次のとおりです:

原始实数张量形状: torch.Size([3, 2])
复数张量形状: torch.Size([3])
复数张量: tensor([ 1.2345+0.5678j, -0.3456+1.2345j,  2.3456-0.9876j])

Pytorch torch 参考手册Pytorch torch リファレンスマニュアル

その他の拡張