評價此頁

torch.isreal#

torch.isreal(input) Tensor#

返回一個新的張量,其中包含布林元素,表示 input 的每個元素是實值還是非實值。所有實值型別都被視為實值。當複數值的虛部為 0 時,它們被視為實值。

引數

input (Tensor) – 輸入張量。

返回

一個布林張量,當 input 是實值時為 True,否則為 False

示例

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