oneflow.isinf¶
-
oneflow.
isinf
()¶ Tests if each element of input is infinite (positive or negative infinity) or not.
- Parameters
input (Tensor) – the input tensor.
- Returns
A boolean tensor that is True where input is infinite and False elsewhere.
Example:
>>> import oneflow as flow >>> flow.isinf(flow.tensor([1, float('inf'), 2, float('-inf'), float('nan')])) tensor([False, True, False, True, False], dtype=oneflow.bool)