antorcha.

Torch Gather

1. antorcha.gather ()

|_+_|

La explicación en el sitio web oficial es agregar valores a lo largo de un eje dado, que es más abstracto. Vayamos directamente al ejemplo.

|_+_|

imagen
dim = 1, por lo que el valor se toma en la dirección de la columna. La primera fila de y es 0 y 1, así que tome los elementos en la columna 0 y la columna 1 de la primera fila de x, y la segunda fila de y es 1 y 2, y luego tome la primera columna y la columna 2 de la segunda. fila de x elemento. Esto obtendrá un nuevo tensor.
Echemos un vistazo a la situación cuando dim = 0.



|_+_|

imagen
Después de leer el resultado, creo que todos lo entienden sin decirlo. El valor de y es 0, 1, 1, por lo que puede tomar los elementos de fila 0, 1 y 1 de cada columna de x.