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]])