Функция torch.isin в PyTorch
Справочное руководство Pytorch torch
torch.isinЭто функция в PyTorch, используемая для проверки того, находятся ли элементы в другом тензоре. Она возвращает булев тензор, указывающий, присутствует ли каждый элемент первого тензора во втором тензоре.
Определение функции
torch.isin(elements, test_elements, assume_unique=False, invert=False)
Примеры использования
Пример
import torch
# Создайте два тензора
elements = torch.tensor([1, 2, 3, 4])
test_elements = torch.tensor([2, 4])
# Проверить, находятся ли elements в test_elements
result = torch.isin(elements, test_elements)
print(result)
# Создайте два тензора
elements = torch.tensor([1, 2, 3, 4])
test_elements = torch.tensor([2, 4])
# Проверить, находятся ли elements в test_elements
result = torch.isin(elements, test_elements)
print(result)
Результат вывода:
tensor([False, True, False, True])
Другие расширения