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