torch.isinf

torch.isinf(input) → Tensor

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

注意

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

引數

input (Tensor) – 輸入張量。

返回

一個布林型張量，在 input 是無窮大的位置為 True，否則為 False

示例

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