torch.isreal

torch.isreal(input) → 張量

返回一個新張量，其布林元素表示 input 中的每個元素是否為實數值。所有實數值型別都被認為是實數。複數值在其虛部為 0 時被認為是實數。

引數

input (張量) – 輸入張量。

返回

一個布林張量，當 input 中對應元素為實數時為 True，否則為 False

示例

>>> torch.isreal(torch.tensor([1, 1+1j, 2+0j]))
tensor([True, False, True])