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