PyTorch torch.is_complex 関数
Pytorch torch リファレンスマニュアルtorch.is_complexPyTorch において、テンソルのデータ型が複素数型かどうかを確認するための関数です。
関数定義
torch.is_complex(input)
使用例
例
import torch
# 実数テンソルをチェック
x = torch.tensor([1.0, 2.0, 3.0])
print("実数:", torch.is_complex(x))
# 複素数テンソルをチェック
y = torch.tensor([1+2j, 3+4j])
print("複素数:", torch.is_complex(y))
# 実数テンソルをチェック
x = torch.tensor([1.0, 2.0, 3.0])
print("実数:", torch.is_complex(x))
# 複素数テンソルをチェック
y = torch.tensor([1+2j, 3+4j])
print("複素数:", torch.is_complex(y))
出力結果は次のとおりです：
实数: False 复数: True
その他の拡張