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 |
Constantes
Cadena | OP_NOMBRE | El nombre de esta operación, como lo conoce el motor central de TensorFlow. |
Métodos públicos
Salida <T> | como salida () Devuelve el identificador simbólico del tensor. |
Opciones estáticas de recopilación | loteDims (loteDims largo) |
estático <T extiende TType > Reunir <T> | crear ( alcance alcance , parámetros de operando <T>, índices de operando <? extiende TNumber >, eje de operando <? extiende TNumber >, opciones... 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
Constantes
Cadena final estática pública OP_NAME
El nombre de esta operación, como lo conoce el motor central de TensorFlow.
Métodos públicos
Salida pública <T> como Salida ()
Devuelve el identificador simbólico del 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.
public static Gather <T> create (Ámbito de alcance , Operando <T> parámetros, Operando <? extiende TNumber > índices, Operando <? extiende TNumber > eje, 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:]`.