Function _check_non_finite

imperative/python/megengine/functional/math.py:1004–1021

r"""Check whether input contains infinite or nan value. Args: inps: tensors to be checked. Returns: a int32 scalar tensor, 0 for False and 1 for True.

(inps: Iterable[Tensor], scale=1.0)

def _check_non_finite(inps: Iterable[Tensor], scale=1.0) -> Tensor:
    r"""Check whether input contains infinite or nan value.

    Args:
        inps: tensors to be checked.

    Returns:
        a int32 scalar tensor, 0 for False and 1 for True.
    """
    if isinstance(inps, Tensor):
        inps = [inps]
    op = builtin.CheckNonFinite(scale=scale)
    oups = apply(op, *inps)
    out = oups[-1]
    for i in range(len(inps)):
        inps[i]._reset(oups[i])

    return out