Python PyTorch isfinite用法及代码示例

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

用法:

torch.isfinite(input) → Tensor

参数：

input(Tensor) -输入张量。

返回：

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

返回一个带有布尔元素的新张量，表示每个元素是否为finite。

当实数值不是 NaN、负无穷或无穷大时，它们是有限的。当复数值的实部和虚部都是有限的时，复数值是有限的。

例子：

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