Gather

reunión de clase final pública

Reúna cortes del eje `params` `eje` de acuerdo con `í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[: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ímites, 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

Cuerda OP_NOMBRE El nombre de esta operación, como lo conoce el motor principal de TensorFlow

Métodos públicos

Salida <T>
como salida ()
Devuelve el identificador simbólico del tensor.
Gather.Options estáticas
batchDims (long batchDims)
estático <T extiende TType > Reúne <T>
crear ( Á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.
Salida <T>
salida ()
Valores de `params` recopilados a partir de índices proporcionados por `indices`, con la 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 principal de TensorFlow

Valor constante: "Reunir V2"

Métodos públicos

salida pública <T> como salida ()

Devuelve el identificador simbólico del tensor.

Las entradas de 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.Options batchDims (long batchDims)

public static Gather <T> create ( Scope scope, 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 a partir del cual recopilar valores. Debe tener al menos el rango `eje + 1`.
índices Tensor í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. Admite índices negativos.
opciones lleva valores de atributos opcionales
Devoluciones
  • una nueva instancia de Gather

salida pública <T> salida ()

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