Функция PyTorch torch.index_select
Справочное руководство Pytorch torch
torch.index_selectЭто функция в PyTorch, используемая для выбора элементов, соответствующих индексам вдоль указанного измерения.
Определение функции
torch.index_select(input, dim, index)
Пример использования
Пример
import torch
x = torch.randn(4, 5)
# Выбрать строки 0 и 2
indices = torch.tensor([0, 2])
result = torch.index_select(x, dim=0, index=indices)
print("Форма результата:", result.shape)
x = torch.randn(4, 5)
# Выбрать строки 0 и 2
indices = torch.tensor([0, 2])
result = torch.index_select(x, dim=0, index=indices)
print("Форма результата:", result.shape)
Другие расширения