PyTorch torch.isreal 関数
torch.isrealPyTorch で実数かどうかをチェックするための関数です。各要素について、実数(複素数ではない)の場合にTrueとなるブールテンソルを返します。
関数定義
torch.isreal(input)
使用例
例
import torch
# 実数と複素数を含むテンソルを作成
x = torch.tensor([1.0, 2+3j, 4.0, 5+1j])
# 実数かどうかをチェック
result = torch.isreal(x)
print(result)
# 実数と複素数を含むテンソルを作成
x = torch.tensor([1.0, 2+3j, 4.0, 5+1j])
# 実数かどうかをチェック
result = torch.isreal(x)
print(result)
出力結果は次のとおりです。
tensor([True, False, True, False])
その他の拡張機能
Pytorch torch リファレンスマニュアル