Python PyTorch isinf用法及代码示例

本文简要介绍python语言中 torch.isinf 的用法。

用法:

torch.isinf(input) → Tensor

参数：

input(Tensor) -输入张量。

返回：

一个布尔张量，在 input 是无限的地方为真，在其他地方为假

测试input 的每个元素是否为无穷大(正无穷或负无穷)。

注意

当复数的实部或虚部是无限的时，复数值是无限的。

例子：

>>> torch.isinf(torch.tensor([1, float('inf'), 2, float('-inf'), float('nan')]))
tensor([False,  True,  False,  True,  False])