PyTorch torch.view_as_real 関数
torch.view_as_realPyTorch で複素数テンソルのビューを実数テンソルに変換するための関数です。各複素数値(実部と虚部)を2つの個別の値に分割し、最後の次元には2つの要素が含まれます。
関数定義
torch.view_as_real(input)
使用例
例
import torch
# 複素数テンソルを作成
x = torch.tensor([1+2j, 3+4j, 5+6j])
# 実数ビューに変換
y = torch.view_as_real(x)
print("元の複素数テンソルの形状:", x.shape)
print("実数テンソル形状:", y.shape)
print("実数テンソル:")
print(y)
# 複素数テンソルを作成
x = torch.tensor([1+2j, 3+4j, 5+6j])
# 実数ビューに変換
y = torch.view_as_real(x)
print("元の複素数テンソルの形状:", x.shape)
print("実数テンソル形状:", y.shape)
print("実数テンソル:")
print(y)
出力結果は:
原始复数张量形状: torch.Size([3])
实数张量形状: torch.Size([3, 2])
实数张量:
tensor([[1., 2.],
[3., 4.],
[5., 6.]])
その他の拡張
Pytorch torch リファレンスマニュアル