评价此页

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