Advertencia: esta API está obsoleta y se eliminará en una versión futura de TensorFlow una vez que el reemplazo sea ​​estable.

Gather

Clase public final Gather

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

`índices` debe ser un tensor entero de cualquier dimensión (generalmente 0-D o ​​1-D). Produce un tensor de salida con forma `params.shape [: eje] + indices.shape [batch_dims:] + params.shape [eje + 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 del límite, se devuelve un error. En la GPU, si se encuentra un índice fuera del 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 Recopilar.Opciones Los atributos opcionales para Gather

Métodos públicos

Salida <T>
asOutput ()
Devuelve el identificador simbólico de un tensor.
estáticas Gather.Options
batchDims (Long batchDims)
estática <T, U se extiende Número, V se extiende Número> Reunir <T>
crear ( Alcance alcance, operando <T> params, operando <U> índices, operando <V> eje, Opciones ... Opciones)
Método de fábrica para crear una clase que envuelva una nueva operación Gather.
Salida <T>
salida ()
Valores de "params" obtenidos de índices dados por "índices", con forma "params.shape [: axis] + indices.shape + params.shape [axis + 1:]".

Métodos heredados

Métodos públicos

pública de salida <T> asOutput ()

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.

estáticas pública Gather.Options batchDims (Long batchDims)

public static Reunir <T> crear ( Alcance alcance, operando <T> params, operando <U> índices, operando <V> eje, Opciones ... Opciones)

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

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

pública de salida <T> salida ()

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