torch.isfinite¶
- torch.isfinite(input) Tensor¶
傳回一個新的張量，其布林元素表示每個元素是否為 finite (有限)。
當實數值不是 NaN、負無限大或無限大時，它們是有限的。當複數的實部和虛部都有限時，它們是有限的。
- 參數
input (Tensor) – 輸入張量。
- 傳回
一個布林張量，在
input為有限值時為 True，其他地方為 False。
範例
>>> torch.isfinite(torch.tensor([1, float('inf'), 2, float('-inf'), float('nan')])) tensor([True, False, True, False, False])
