评价此页

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])