PyTorch: Función de corte index_select()

La función index_select() tiene dos propósitos.

La primera es pasar la función de división como parámetro a index_select(),

La otra es llamar a la función index_select() incorporada del tensor.

La función index_select() realiza esta operación en la entrada del tensor, dividiendo el rango especificado por el índice entre sus dimensiones.

Parámetros:

Entrada: tensor sobre el que se va a operar

dim: dimensión

índice: una dimensión que representa el rango de subíndices del índice Tensor

Por ejemplo

la salida es

Aquí la dimensión tenue comienza a contar desde 0, por lo que b representa la fila (es decir, la fila) de la dimensión 0 con superíndice y subíndice 1; c representa las columnas de la dimensión 1 con los subíndices 1 y 3 (es decir, columnas).