Fonction torch.index_select de PyTorch

Pytorch torch 参考手册Manuel de référence Pytorch torch

torch.index_selectC'est une fonction de PyTorch utilisée pour sélectionner les éléments correspondant aux indices le long d'une dimension spécifiée.

Définition de fonction

torch.index_select(input, dim, index)

Exemple d'utilisation

Exemple

import torch

x = torch.randn(4, 5)

# Sélectionner les lignes 0 et 2
indices = torch.tensor([0, 2])
result = torch.index_select(x, dim=0, index=indices)

print("Forme du résultat :", result.shape)

Pytorch torch 参考手册Manuel de référence Pytorch torch

Autres extensions