Gather

reunión pública de clase final

Reúna cortes del eje `params` `axis` según los `índices`.

Los "índices" deben ser un tensor entero de cualquier dimensión (normalmente 0-D o ​​1-D). Produce un tensor de salida con forma `params.shape[:axis] + indices.shape[batch_dims:] + params.shape[axis + 1:]` donde: 

# Scalar indices (output is rank(params) - 1).
     output[a_0, ..., a_n, b_0, ..., b_n] =
       params[a_0, ..., a_n, indices, b_0, ..., b_n]
 
     # Vector indices (output is rank(params)).
     output[a_0, ..., a_n, i, b_0, ..., b_n] =
       params[a_0, ..., a_n, indices[i], b_0, ..., b_n]
 
     # Higher rank indices (output is rank(params) + rank(indices) - 1).
     output[a_0, ..., a_n, i, ..., j, b_0, ... b_n] =
       params[a_0, ..., a_n, indices[i, ..., j], b_0, ..., b_n]

Tenga en cuenta que en la CPU, si se encuentra un índice fuera de límite, se devuelve un error. En GPU, si se encuentra un índice fuera de límite, se almacena un 0 en el valor de salida correspondiente.

Consulte también `tf.batch_gather` y tf.gather_nd .

Clases anidadas

clase Reunir.Opciones Atributos opcionales para Gather

Métodos públicos

Salida <T>
como salida ()
Devuelve el identificador simbólico de un tensor.
Opciones estáticas de recopilación
loteDims (loteDims largo)
estático <T, U extiende el número, V extiende el número> Reunir <T>
crear (alcance del alcance , parámetros del operando <T>, índices del operando <U>, eje del operando <V>, opciones ... )
Método de fábrica para crear una clase que envuelve una nueva operación Gather.
Salida <T>
producción ()
Valores de `params` recopilados de índices dados por `indices`, con forma `params.shape[:axis] + indices.shape + params.shape[axis + 1:]`.

Métodos heredados

Métodos públicos

Salida pública <T> como Salida ()

Devuelve el identificador simbólico de un tensor.

Las entradas a las operaciones de TensorFlow son salidas de otra operación de TensorFlow. Este método se utiliza para obtener un identificador simbólico que representa el cálculo de la entrada.

público estático Gather.Options loteDims (loteDims largos)

public static Gather <T> create (alcance del alcance , parámetros del operando <T>, índices del operando <U>, eje del operando <V>, opciones... opciones)

Método de fábrica para crear una clase que envuelve una nueva operación Gather.

Parámetros
alcance alcance actual
parámetros El tensor del que se obtienen valores. Debe tener al menos el rango "eje + 1".
índices Tensor de índice. Debe estar en el rango `[0, params.shape[axis])`.
eje El eje en `params` para recopilar `índices`. El valor predeterminado es la primera dimensión. Soporta índices negativos.
opciones lleva valores de atributos opcionales
Devoluciones
  • una nueva instancia de Gather

Salida pública <T> salida ()

Valores de `params` recopilados de índices dados por `indices`, con forma `params.shape[:axis] + indices.shape + params.shape[axis + 1:]`.