torch.isfinite¶
- torch.isfinite(input) Tensor¶
返回一個布林張量，其元素表示對應輸入元素是否是有限的。
實數在不是 NaN、負無窮大或無窮大時是有限的。複數在其實部和虛部都是有限的時是有限的。
- 引數
input (Tensor) – 輸入張量。
- 返回值
一個布林張量，當
input中的元素是有限時為 True，否則為 False。
示例
>>> torch.isfinite(torch.tensor([1, float('inf'), 2, float('-inf'), float('nan')])) tensor([True, False, True, False, False])
