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