Function in1d

monai/transforms/utils_pytorch_numpy_unification.py:87–91

`np.in1d` with equivalent implementation for torch.

(x, y)

def in1d(x, y):
    """`np.in1d` with equivalent implementation for torch."""
    if isinstance(x, np.ndarray):
        return np.isin(x, y)
    return (x[..., None] == torch.tensor(y, device=x.device)).any(-1).view(-1)


def clip(a: NdarrayOrTensor, a_min, a_max) -> NdarrayOrTensor: