torch.isinf

torch.isinf(input) → Tensor

測試 input 的每個元素是否為無限大（正或負無限大）。

注意

當複數的實部或虛部為無限大時，該複數為無限大。

參數

input ( Tensor ) – 輸入張量 (input tensor)。

返回 (Returns)

一個布林 (Boolean) 張量，當 input 為無限大 (infinite) 時為 True，其他情況為 False。

範例 (Example)

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