PyTorch torch.masked_select Функция
Справочное руководство PyTorch torch
torch.masked_selectЭто функция в PyTorch, используемая для выбора элементов на основе булевой маски.
Определение функции
torch.masked_select(input, mask)
Пример использования
Пример
import torch
x = torch.tensor([1, 2, 3, 4, 5])
mask = torch.tensor([True, False, True, False, True])
# Выбрать элементы на основе маски
result = torch.masked_select(x, mask)
print(result)
x = torch.tensor([1, 2, 3, 4, 5])
mask = torch.tensor([True, False, True, False, True])
# Выбрать элементы на основе маски
result = torch.masked_select(x, mask)
print(result)
Результат вывода:
tensor([1, 3, 5])
Другие расширения