PyTorch torch.imag 関数
torch.imagは PyTorch で複素数テンソルの虚部(imaginary part)を取得するための関数です。
関数定義
torch.imag(input)
使用例
実例
import torch
# 複素数テンソルを作成
x = torch.complex(torch.tensor([1.0, 2.0]), torch.tensor([3.0, 4.0]))
# 虚部を取得
y = torch.imag(x)
print("虚部:", y)
# 複素数テンソルを作成
x = torch.complex(torch.tensor([1.0, 2.0]), torch.tensor([3.0, 4.0]))
# 虚部を取得
y = torch.imag(x)
print("虚部:", y)
出力結果は:
虚部: tensor([3., 4.])
その他の拡張
Pytorch torch リファレンスマニュアル