PyTorch torch.isreal 関数


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

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)

出力結果は次のとおりです。

tensor([True, False, True, False])

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

その他の拡張機能