torch.isin

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

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

Примечание

Лишь один из аргументов elements и test_elements может быть скаляром.

Параметры:

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

Возвращает:

Логический тензор той же формы, что и elements: для элементов из test_elements он содержит True, а для остальных — False

Пример

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