PyTorch torch.imag 関数


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

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)

出力結果は:

虚部: tensor([3., 4.])

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

その他の拡張