Función PyTorch torch.index_select

Pytorch torch 参考手册Manual de referencia de Pytorch torch

torch.index_selectEs una función en PyTorch que selecciona los elementos correspondientes a los índices a lo largo de una dimensión especificada.

Definición de la función

torch.index_select(input, dim, index)

Ejemplo de uso

Ejemplo

import torch

x = torch.randn(4, 5)

# Seleccionar las filas 0 y 2
indices = torch.tensor([0, 2])
result = torch.index_select(x, dim=0, index=indices)

print(Forma del resultado:, result.shape)

Pytorch torch 参考手册Manual de referencia de Pytorch torch

Otras extensiones