torch.isin

torch.isin(elements, test_elements, *, assume_unique=False, invert=False) → Tensor

Проверяет, является ли каждый элемент elements частью test_elements. Возвращает булевый тензор такой же формы, как elements, в котором True соответствует элементам, присутствующим в test_elements, а False — в противном случае.

Примечание

Один из elements или test_elements может быть скаляром, но не оба.

Параметры:

• elements (Тензор или Скаляр) – Входные элементы
• test_elements (Тензор или Скаляр) – Значения, с которыми проверяются каждый входной элемент
• assume_unique (bool, необязательно) – Если True, предполагает, что оба elements и test_elements содержат уникальные элементы, что может ускорить вычисления. По умолчанию: False
• invert (bool, необязательно) – Если True, инвертирует булевый возвращаемый тензор, что приводит к True для элементов, не присутствующих в test_elements. По умолчанию: False

Возвращаемое значение:

Булевый тензор такой же формы, как elements, в котором True соответствует элементам, присутствующим в test_elements, а False — в противном случае

Пример

>>> torch.isin(torch.tensor([[1, 2], [3, 4]]), torch.tensor([2, 3]))
tensor([[False,  True],
        [ True, False]])