Function isfinite

monai/transforms/utils_pytorch_numpy_unification.py:338–342

`np.isfinite` with equivalent implementation for torch.

(x: NdarrayOrTensor)

def isfinite(x: NdarrayOrTensor) -> NdarrayOrTensor:
    """`np.isfinite` with equivalent implementation for torch."""
    if not isinstance(x, torch.Tensor):
        return np.isfinite(x)  # type: ignore
    return torch.isfinite(x)


def searchsorted(a: NdarrayTensor, v: NdarrayOrTensor, right=False, sorter=None, **kwargs) -> NdarrayTensor: