Python PyTorch isreal用法及代碼示例

本文簡要介紹python語言中 torch.isreal 的用法。

用法:

torch.isreal(input) → Tensor

參數：

input(Tensor) -輸入張量。

返回：

一個布爾張量，在 input 是真實的地方為真，在其他地方為假

返回一個帶有布爾元素的新張量，表示 input 的每個元素是否為實值。所有實值類型都被認為是實數。當複數值的虛部為 0 時，複數值被認為是實數。

例子：

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