torch.isreal

torch.isreal(input) → 张量

返回一个包含布尔元素的新张量，表示input中的每个元素是否为实数值。 所有实数类型都被认为是实数。当复数值的虚部为0时，它们被认为是实数。

Parameters

输入 (张量) – 输入张量。

Returns

一个布尔张量，在 input 为实数的地方为 True，其他地方为 False

示例：

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